一.问题描述
WARNING ⚠️ Box and segment counts should be equal, but got len(segments) = 102, len(boxes) = 105. To resolve this only boxes will be used and all segments will be removed. To avoid this please supply either a detect or segment dataset, not a detect-segment mixed dataset.
大概意思就是标注文件中有segment和bbox的数量对不上,就是数据集中不全是segment标注。
二.解决办法
labelme标注的数据集可以有多种格式,包括但不限于
1.polygon(实例分割的标准格式)
2.rectangle(bbox框格式)
3.circle(圆圈格式)
如果数据集中包含rectangle等标注格式,不加修改直接转yolo格式进行训练的话,会出现数据集错误,产生如上报错信息
2.1 筛选json文件中的标注格式
import os
import json
import cv2
import numpy as np
import shutil
# json目录
fileDir = '/home/fc/sft/Paper_Detection/python/afan/yolov8/dataset/dataset_lingjian3/json'
output_cropped_img_dir = '/home/afan/Desktop/down/new'
# 使用os.listdir()来列出目录中的所有文件
file_names =