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