(detectron之源码解读)/detectron/lib/datasets/
2018-03-19 13:25
253 查看
1. dataset_catalog.py
当我们需要训练自己的数据时(参考:http://blog.csdn.net/meccaendless/article/details/79457330),我们需要对该文件的内容进行修改,在我用自己的数据进行训练的时候,我采用的是VOC2007的数据结构,因此我对该文件中的下面部分进行了修改: 'voc_2007_trainval': { IM_DIR: _DATA_DIR + '/VOC2007/JPEGImages', ANN_FN: _DATA_DIR + '/VOC2007/annotations/voc_2007_trainval.json', DEVKIT_DIR: _DATA_DIR + '/VOC2007/VOCdevkit2007' }, 'voc_2007_test': { IM_DIR: _DATA_DIR + '/VOC2007/JPEGImages', ANN_FN: _DATA_DIR + '/VOC2007/annotations/voc_2007_test.json', DEVKIT_DIR: _DATA_DIR + '/VOC2007/VOCdevkit2007' },
修改为下面的形式:
'voc_2007_trainval': { IM_DIR: _DATA_DIR + '/VOC2007/JPEGImages', ANN_FN: _DATA_DIR + '/VOC2007/annotations/pascal_trainval2007.json', DEVKIT_DIR: _DATA_DIR + '/VOC2007/VOCdevkit2007' }, 'voc_2007_test': { IM_DIR: _DATA_DIR + '/VOC2007/JPEGImages', ANN_FN: _DATA_DIR + '/VOC2007/annotations/pascal_test2007.json', DEVKIT_DIR: _DATA_DIR + '/VOC2007/VOCdevkit2007' },
2. dummy_datasets.py
当我们训练自己的数据时,该文件并非一定去要修改,但是当最后用infer_simple.py测试模型时,需要修改里面的classes的内容,否则图片上的标签都是COCO数据集的标签了。ps:不要忘记了'__background__'
当我们需要训练自己的数据时(参考:http://blog.csdn.net/meccaendless/article/details/79457330),我们需要对该文件的内容进行修改,在我用自己的数据进行训练的时候,我采用的是VOC2007的数据结构,因此我对该文件中的下面部分进行了修改: 'voc_2007_trainval': { IM_DIR: _DATA_DIR + '/VOC2007/JPEGImages', ANN_FN: _DATA_DIR + '/VOC2007/annotations/voc_2007_trainval.json', DEVKIT_DIR: _DATA_DIR + '/VOC2007/VOCdevkit2007' }, 'voc_2007_test': { IM_DIR: _DATA_DIR + '/VOC2007/JPEGImages', ANN_FN: _DATA_DIR + '/VOC2007/annotations/voc_2007_test.json', DEVKIT_DIR: _DATA_DIR + '/VOC2007/VOCdevkit2007' },
修改为下面的形式:
'voc_2007_trainval': { IM_DIR: _DATA_DIR + '/VOC2007/JPEGImages', ANN_FN: _DATA_DIR + '/VOC2007/annotations/pascal_trainval2007.json', DEVKIT_DIR: _DATA_DIR + '/VOC2007/VOCdevkit2007' }, 'voc_2007_test': { IM_DIR: _DATA_DIR + '/VOC2007/JPEGImages', ANN_FN: _DATA_DIR + '/VOC2007/annotations/pascal_test2007.json', DEVKIT_DIR: _DATA_DIR + '/VOC2007/VOCdevkit2007' },
2. dummy_datasets.py
当我们训练自己的数据时,该文件并非一定去要修改,但是当最后用infer_simple.py测试模型时,需要修改里面的classes的内容,否则图片上的标签都是COCO数据集的标签了。ps:不要忘记了'__background__'
相关文章推荐
- (detectron之源码解读)/detectron/lib/core/config.py
- Hypertable源码解读之Hypertable.lib目录
- redux-applyMiddleware源码解读
- 从源码解读Mysql 5.7性能和数据安全性的提升
- SDWebImage源码解读之干货大总结
- 全方位解读Android多媒体框架源码
- 通过opendaylight源码解读控制器与openflow交换机建立过程
- Java HashMap 核心源码解读
- java.io.BufferedWriter API 以及源码解读
- Flask源码解读 <2> --- 请求上下文和request对象
- facebook presto源码解读(part2)
- java.nio.ByteBuffer源码解读
- Docker网络详解及pipework源码解读与实践
- AMPS:数据库访问模块源码解读
- 解读linux内核源码的入门方法 [转]
- AFNetworking 3.0 源码解读(三)之 AFURLRequestSerialization
- spring core源码解读之ASM4用户手册翻译之一asm简介
- 编译Android源码 解决:[out/host/linux-x86/obj/SHARED_LIBRARIES/libclearsilver-jni_intermediates/j_neo_util.o] 错误 1
- String源码分析,解读面试题
- ThreadLocal使用及源码解读