8.5 RCNN系列推荐阅读

RCNN系列的目标检测方法是经典的Two Stage方法,优先推荐同学们阅读该系列的论文。此外,为了帮助大家理解,下文为大家推荐部分内容较好的中文博客。

8.6 RCNN相关代码参考

目前RCNN系列最为经典,同时也是使用最多的网络是何凯明提出的Mask R-CNN, 在此处为大家提供论文附带的代码以及在此基础上Facebook改进的代码。

两者使用的均是Pytorch框架,其中Detectron2使用的Pytorch版本更新。此外,如果对网络结构不加更改,可以直接使用TorchVision中Mask R-CNN或者Faster R-CNN。

TorchVision

对于对Keras/TF这类静态图框架有偏好的同学,这里也提供一份TF和Keras结合实现的Mask R-CNN代码。

Mask R-CNN - Keras implemented

[ ]: