faster RCNN代码阅读笔记

faster RCNN代码阅读笔记

文章目录

  • faster RCNN代码阅读笔记
    • 本项目直接fork了jwyang/faster-rcnn.pytorch,旨在学习faster RCNN
    • 主网络模型构建部分(vgg16.py和faster_RCNN.py)

项目链接 https://github.com/qqsuhao/faster-rcnn.pytorch

本项目直接fork了jwyang/faster-rcnn.pytorch,旨在学习faster RCNN

主网络模型构建部分(vgg16.py和faster_RCNN.py)

faster RCNN代码阅读笔记_第1张图片

faster RCNN代码阅读笔记_第2张图片
对照上面这张图和faster RCNN的网络结构,可以更加清楚地了解faster RCNN的结构。在下面这张图中,我标注了代码中各个模块对应的对象名和类名,方便读者将结构图与代码进行一一对应。这里没有标注ROI_pooling在代码中的名称,主要是因为代码中有三种不同的方式进行ROI_pooling,具体还需要之后进一步详细了解。

你可能感兴趣的:(faster,RCNN,人工智能,python)