文章目錄 [隱藏]

• 1 什么是 Mask-RCNN
• 2 PyTorch 實(shí)現(xiàn) Mask-RCNN
  • 2.1 文件描述
  • 2.2 Mask-RCNN 模型

什么是 Mask-RCNN

Mask-RCNN 來自于 Kaiming He 的一篇論文，通過在 Faster-RCNN 的基礎(chǔ)上添加一個(gè)分支網(wǎng)絡(luò)，在實(shí)現(xiàn)目標(biāo)檢測的同時(shí)，把目標(biāo)像素分割出來。論文地址。

PyTorch 實(shí)現(xiàn) Mask-RCNN

這是一個(gè)用 PyTorch 實(shí)現(xiàn)的 Mask-RCNN 網(wǎng)絡(luò)，Github 地址為：https://github.com/wannabeOG/Mask-RCNN。

文件描述

model.py 包含 ResNet 和 FPN 的模型

nms 和 RoiAlign 是從 Robb Girshick 的 faster RCNN 實(shí)現(xiàn)中借鑒的

本代碼實(shí)現(xiàn)了 Focal loss

Mask-RCNN 模型

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

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。