mirror of
https://gitee.com/nanjing-yimao-information/ieemoo-ai-gift.git
synced 2025-08-18 21:30:25 +00:00
Merge branch 'master' of https://gitee.com/nanjing-yimao-information/ieemoo-ai-gift
This commit is contained in:
@ -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()
|
||||
|
@ -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
|
||||
# 标注越界修正
|
||||
|
Reference in New Issue
Block a user