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) 编辑 收藏 举报