Faster R-CNN官方公布的代码主要是针对Pascal、COCO数据集的,如果要针对特定任务训练自己的数据集,需要做一些额外的工作。
方式一:自己标注 https://saicoco.github.io/object-detection-4/
https://github.com/ChaoPei/create-voc2007-dataset
方式二:下载外部数据集,通常格式各异,又分两种情况:
- 修改工程代码,适应自己的数据集格式:
- 将自己的数据格式做成VOC2007形式的数据集: