文章目錄 [隱藏]

• 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。

文件描述

model.py 包含 ResNet 和 FPN 的模型

nms 和 RoiAlign 是從 Robb Girshick 的 faster RCNN 實現中借鑒的

本代碼實現了 Focal loss

Mask-RCNN 模型

如果大家有什么建議，可以到原 Github 上進行討論。

總結

以上是生活随笔為你收集整理的PyTorch 实现 Mask-RCNN的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。