YOLOV4模型剪枝后在Xavier开发板上训练自己的数据集
一、标注数据
使用labelImg进行标注,标注方法自行研究,标注完成后生成的是xml后缀的文件,该格式是VOC的标准格式。
二、制作自己的数据集
我这里总共标注训练数据4351张,类别为船(共一类),数据集制作参考yoloV4里面的readme。
import os
import xml.etree.ElementTree as ET
if __name__ == '__main__':
list_file = os.listdir('F:\\155dataSet\\boatAll\\an
原创
2020-06-30 17:14:32 ·
4301 阅读 ·
26 评论