用YOLO对绝缘子数据集进行目标检测
InsulatorDataSet包含绝缘子的图片以及目标和故障的标记。详细内容请看 InsulatorDataSet
将insulatorDataSet2YOLO.py放到数据集目录下即可。insulatorDataSet2YOLO.py会自动将原先的VOC数据集格式转化成成YOLO格式并划分好训练以及验证。yolov8训练该 数据集可以参考ultralytics.com官方文档
- 原始数据集结构
└── InsulatorDataSet
└── images
└── xxx.jpg
└── labels
└── insulator
└── xxx.xml
└── defect.jpg
└── xxx.xml
- YOLO数据集格式
└── InsulatorDataSet_YOLO
└── images
└── train
└── xxx.jpg
└── val
└── xxx.jpg
└── labels
└── train
└── xxx.txt
└── val
└── xxx.txt