PyTorch 实现 Mask-RCNN

文章目录 [隐藏]

• 1 什么是 Mask-RCNN
• 2 PyTorch 实现 Mask-RCNN
  • 2.1 文件描述
  • 2.2 Mask-RCNN 模型

什么是 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 模型

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