Faster-RCNN实战


仅用于个人学习与总结,持续更新中......

1 前言

2 原理篇

3 代码篇

  首先附上本文所用的源码地址:https://github.com/chenyuntc/simple-faster-rcnn-pytorch,该版本的代码非常简洁干练,深刻的感觉到作者的代码功底的深厚,在此对源码作者表示由衷的感激和深深的敬意。

  源码主要目录结构如下:

|- data # 
  - __init__.py
  - dataset.py
  - util.py
  - voc_dataset.py
|- misc # 
  - convert_caffe_pretrain.py
  - demo.jpg
  - train_fast.py
|- model # 构建Faster RCNN网络模型有关的文件
  |- utils
    - __init__.py
    - bbox_tools.py
    - creator_tool.py
  - __init__.py
  - faster_rcnn.py
  - faster_rcnn_vgg16.py
  - region_proposal_network.py
|- utils # 辅助可视化和验证的文件
  - __init__.py
  - array_tool.py
  - config.py
  - eval_tool.py
  - vis_tool.py
| demo.ipynb
| requirements.txt
| train.py # 整个程序的运行文件
| trainer.py # 训练、模型损失函数计算等