第一步份修改的是数据:VOC数据集 VOC0712.py
第二部分是配置:config.py
第三部分:train.py 看情况改吧
第一个bug:使用了自己的voc数据集报错误却是coco的。
Traceback (most recent call last):
File “train.py”, line 1, in
from data import *
File “/home/bitnami/SSD/data/init.py”, line 3, in
from .coco import COCODetection, COCOAnnotationTransform, COCO_CLASSES, COCO_ROOT, get_label_map
File “/home/bitnami/SSD/data/coco.py”, line 75, in
class