目标检测YOLO实战应用案例100讲-基于多尺度特征融合的水下小目标检测方法研究(中)
本案例原始数据存在标注错误的情况,如标注数据中图像长宽为0,本案例将这部分数据剔除掉,不参与训练和测试。将数据集按照9:1的比例进行划分,并生成train.txt和val.txt进行训练import os#生成train.txt和val.txtxml_dir = '/home/aistudio/train-A/box'#标签文件地址img_dir = '/home/aistudio/train-A/image'#图像文件地址。
复制链接