PyTorch 实现 Mask-RCNN

PyTorch入门实战教程

什么是 Mask-RCNN

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

PyTorch 实现 Mask-RCNN

这是一个用 PyTorch 实现的 Mask-RCNN 网络,Github 地址为: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 上进行讨论。

PyTorch入门实战教程

Leave a Reply

Your email address will not be published. Required fields are marked *

返回顶部