PyTorch 实现 Mask-RCNN

柔情只为你懂 2022-02-25 16:10 306阅读 0赞

文章目录 \[[隐藏][Link 1]\]

*  [1 什么是 Mask-RCNN][1 _ Mask-RCNN]
 *  [2 PyTorch 实现 Mask-RCNN][2 PyTorch _ Mask-RCNN]
    
     *  [2.1 文件描述][2.1]
     *  [2.2 Mask-RCNN 模型][2.2 Mask-RCNN]

# 什么是 Mask-RCNN #

Mask-RCNN 来自于 Kaiming He 的一篇论文，通过在 Faster-RCNN 的基础上添加一个分支网络，在实现目标检测的同时，把目标像素分割出来。[论文地址][Link 2]。

[![maskrcnn.jpg][]][maskrcnn.jpg]

# PyTorch 实现 Mask-RCNN #

这是一个用 PyTorch 实现的 Mask-RCNN 网络，Github 地址为：[https://github.com/wannabeOG/Mask-RCNN][https_github.com_wannabeOG_Mask-RCNN]。

## 文件描述 ##

1.  model.py 包含 ResNet 和 FPN 的模型
2.  nms 和 RoiAlign 是从 Robb Girshick 的 faster RCNN 实现中借鉴的
3.  本代码实现了 Focal loss

## Mask-RCNN 模型 ##

![alt text][]

如果大家有什么建议，可以到原 Github 上进行讨论。

[Link 1]: https://www.pytorchtutorial.com/pytorch-mask-rcnn/#
[1 _ Mask-RCNN]: https://www.pytorchtutorial.com/pytorch-mask-rcnn/#_Mask-RCNN
[2 PyTorch _ Mask-RCNN]: https://www.pytorchtutorial.com/pytorch-mask-rcnn/#PyTorch_Mask-RCNN
[2.1]: https://www.pytorchtutorial.com/pytorch-mask-rcnn/#i
[2.2 Mask-RCNN]: https://www.pytorchtutorial.com/pytorch-mask-rcnn/#Mask-RCNN
[Link 2]: https://www.pytorchtutorial.com/goto/https://arxiv.org/pdf/1703.06870.pdf
[maskrcnn.jpg]: /images/20220225/bf7c38ad02cb42e89a888960e9f611a5.png
[https_github.com_wannabeOG_Mask-RCNN]: https://www.pytorchtutorial.com/goto/https://github.com/wannabeOG/Mask-RCNN
[alt text]: /images/20220225/1177a7d27c834a89a5ce83d012a37b6b.png

发表评论取消回复

表情：

评论列表（有 0 条评论，306人围观）

还没有评论，来说两句吧...

相关阅读

相关 Pytorch编译maskRCNN问题：cpp_extension.py:189: UserWarning: Error checking compiler version for cl...

先给出两个报错的内容： d:\\Anaconda3\\envs\\torch\\lib\\site-packages\\torch\\utils\\cpp\_exte...

桃扇骨/ 2024年04月18日 23:51/ 0 赞/ 52 阅读

相关 Pytorch实现CSPDarknet53

import torch import torch.nn.functional as F import torch.nn as nn impo

本是古典何须时尚/ 2023年01月13日 10:58/ 0 赞/ 197 阅读

相关 pytorch实现unet

unet是非常经典的图像分割的网络，因网络结构形似字母U而著称实现起来不是很复杂，代码如下： -- coding: utf-8 -- import to

我会带着你远行/ 2022年10月28日 12:23/ 0 赞/ 268 阅读

相关 StemBlock pytorch实现

本文mark一下peleeent和yolov5中的实现： import time from collections import OrderedDict

柔光的暖阳◎/ 2022年10月10日 06:12/ 0 赞/ 178 阅读

相关 onnx推理maskrcnn

import onnxruntime as rt import cv2 from torchvision.ops.boxes import nms

￡神魔★判官ぃ/ 2022年08月28日 00:37/ 0 赞/ 182 阅读

相关 MaskRCNN-Benchmark(Pytorch版本)训练自己的数据以及避坑指南

一、安装地址：[MaskRCNN-Benchmark(Pytorch版本)][MaskRCNN-Benchmark_Pytorch] 首先要阅读官网说明的[环境要求][L

╰半橙微兮°/ 2022年04月25日 04:52/ 0 赞/ 181 阅读

相关 MaskRCNN-ICCV2017 论文解读

文章: MaskRCNN 作者: Kaiming He, Georgia Gkioxari, Piotr Dollar, Ross Girshick 备注: FAIR,

浅浅的花香味﹌/ 2022年04月17日 00:37/ 0 赞/ 280 阅读

相关 pytorch 实现LSTM

\\ Pytorch基础知识点整理 \\ 梯度下降： __coding:utf-8__ from math import pi impo

客官°小女子只卖身不卖艺/ 2022年03月28日 09:44/ 0 赞/ 261 阅读

相关 Pytorch运行官方maskrcnn-benchmark报错: no instance of function template "THCCeilDiv" matches

Pytorch运行官方maskrcnn-benchmark，问题和官网上的位置不同，我的报错在这个位置，SigmoidFocalLoss\_cuda.cu，报

素颜马尾好姑娘i/ 2022年03月18日 12:52/ 0 赞/ 264 阅读

相关 PyTorch 实现 Mask-RCNN

文章目录 \[[隐藏][Link 1]\] [1 什么是 Mask-RCNN][1 _ Mask-RCNN] [2 PyTorch 实现 Mask-RCNN][2

柔情只为你懂/ 2022年02月25日 16:10/ 0 赞/ 307 阅读