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 # 训练、模型损失函数计算等