0°

PyTorch 实现 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 模型

alt text

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

PyTorch入门实战教程
0 条回复 A 作者 M 管理员
    所有的伟大,都源于一个勇敢的开始!
欢迎您,新朋友,感谢参与互动!欢迎您 {{author}},您在本站有{{commentsCount}}条评论