py-faster-rcnn 源码解析2

然后,我们看一下fast rcnn的代码。

lib/datasets
1.factory.py
给该文件一个数据集名字,产出一个数据集(imdb)
2.imdb.py
定义了imdb类和方法
3.pascal_voc.py
定义了一个pascol _voc的类,该类继承自imdb,用于负责数据交互部分。
4.mcg_munge.py

5.coco.py
定义了coco类,该类继承自imdb
6.ds_utils.py
用于支持MS COCO数据集的文件,将数据集格式转变
7.voc_eval.py

lib/fast_rcnn/
1.bbox_transform.py
bounding box 的转换
2.nms_wrapper.py
nms的封装
3.test.py
测试fast rcnn
4.train.py
训练fast rcnn
5.config.py
这里面的东西系不能改,只能override

lib/roi_data_layer/
1.layer.py
为原模型加一个roi层
2.minibatch.py

3.roidb.py
继承自imdb

posted on 2017-10-24 23:06  MissSimple  阅读(775)  评论(0编辑  收藏  举报

导航