已经得到了如下格式的VOC数据集
1 修改类别
-
文件路径:exps/example/yolox_voc/yolox_voc_s.py,本次使用的类别有2类,所以将num_classes修改为2。
-
打开yolox/data/datasets/voc_classes.py文件,修改为自己的类别名:
2 修改数据集目录
-
文件路径:exps/example/yolox_voc/yolox_voc_s.py,data_dir修改为“./data/VOCdevkit”,image_sets删除2012的,最终结果如下:
-
接着往下翻,修改test的路径,如下图:
-
打开yolox/data/datasets/voc.py,这里面有个错误。画框位置,将大括号的“%s”去掉,否则验证的时候一直报找不到文件的错误。
-
如果标注文件中,bbox的四个参数不是int的话(如float),yolox/evaluators/voc_eval.py中,将int全部改为float,如下图:
obj_struct["bbox"] = [ float(bbox.find("xmin").text), float(bbox.find("ymin").text), float(bbox.find("xmax").text), float(bbox.find("ymax").text), ]