paper:Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks
pytorch-faster-rcnn github-----pytorch-faster-rcnn
1. 简介
faster rcnn是何凯明等大神在2015年提出目标检测算法, 一直作为经典的两阶段检测算法。该算法在2015年的ILSVRV和COCO竞赛中获得多项第一。本检测框架是基于pytorch实现的faster rcnn源码。详细的论文细节请参考Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks
本检测框架有如下特点:
- 纯python代码, 完全是基于pytorch,不需要额外build
- 个性化配置,超参数只需要在配置文件中进行配置即可进行配置
- 有多个backbone训练网络. 包括vgg, resnet-fpn, mobilenet, high resolution net(HRNet)等
- 只需修改配置参数即可进行不同模型的训练,对比不同超参数
- 内存占用少,vgg16只需3G左右
2. 安装
2.1 Prerequisites
- Python 2.7 or 3.5