This commit is contained in:
2025-06-21 13:42:44 +08:00
6 changed files with 2174 additions and 5 deletions

View File

@ -7,6 +7,7 @@ parser = argparse.ArgumentParser()
parser.add_argument('--img_path', default='/home/lc/data_center/gift/v2/images', type=str,
help='input xml label path') # 图片存放地址
# 数据集的划分地址选择自己数据下的ImageSets/Main
# parser.add_argument('--txt_path', default='/home/lc/data_center/gift/yolov10_data/Main', type=str, help='output txt label path')
parser.add_argument('--txt_path', default='/home/lc/data_center/gift/yolov10_data/Main', type=str,
help='output txt label path')
opt = parser.parse_args()

View File

@ -5,7 +5,7 @@ from os import getcwd
sets = ['train', 'val', 'test']
classes = ['tag', 'bandage']
classes = ['tag', 'bandage', 'word', 'package']
def convert(size, box):
@ -51,8 +51,9 @@ def convert_annotation(image_id, imgname_list, label_path, Annotation_path, imag
cls_id = classes.index(cls)
xmlbox = obj.find('bndbox')
b = (
float(xmlbox.find('xmin').text), float(xmlbox.find('xmax').text), float(xmlbox.find('ymin').text),
float(xmlbox.find('ymax').text))
float(xmlbox.find('xmin').text), float(xmlbox.find('xmax').text),
float(xmlbox.find('ymin').text),
float(xmlbox.find('ymax').text))
b1, b2, b3, b4 = b
# 标注越界修正