解决YOLOv5_obb数据集检查出错:
WARNING:ignoring corrupt image/label
解决方法:找到datasets.py文件:
File "D:\pre_yolov5_csl\utils\datasets.py", line 132, in __init__
在 def verify_image_label(args):函数中
for label in labels:
if label[-1] == "2": # diffcult
continue
cls_id = cls_name_list.index(label[8])
l_.append(np.concatenate((cls_id, label[:8]), axis=None))
l = np.array(l_, dtype=np.float32)
添加如下语句,并在下文中用cls_list代替即可。
for label in labels:
if label[-1] == "2": # diffcult
continue
# 添加
cls_list = list(cls_name_list.values())
cls_id = cls_list.index(label[8])
l_.append(np.concatenate((cls_id, label[:8]), axis=None))
l = np.array(l_, dtype=np.float32)