simple-faster-rcnn-pytorch 训练自己的数据 (pascal voc格式)
项目地址:simple-faster-rcnn-pytorch
修改脚本:
将voc_dataset.py和vis_tool.py里面的VOC_BBOX_LABEL_NAMES类别替换成自己标签中类别名称,必须和xml文件里面的标签名字一致,否则会直接跳过这个样本。
修改config.py中的config.py中的voc_data_dir路径,改成自己的路径.
修改faster_rcnn_...
原创
2019-04-02 14:41:59 ·
2411 阅读 ·
8 评论