lichen/ieemoo-ai-gift
1
0
Fork 0
mirror of https://gitee.com/nanjing-yimao-information/ieemoo-ai-gift.git synced 2025-08-23 23:50:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

赠品测试

Browse Source
This commit is contained in:
lee
2025-06-21 13:37:38 +08:00
parent e474ab5f9f
commit aa7ea8ecf8
5 changed files with 2174 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
datasets_preprocess/voc_label_.py
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
# 标注越界修正
@ -76,9 +77,9 @@ def convert_annotation(image_id, imgname_list, label_path, Annotation_path, imag
# abs_path = os.getcwd()
image_path = '/home/lc/data_center/gift/ori_image/images/' # img实际存放地址
Annotation_path = '/home/lc/data_center/gift/ori_image/xmls/' # xml实际存放地址
label_path = '/home/lc/data_center/gift/ori_image/labels/' # 保存路径
image_path = '/home/lc/data_center/gift/v2/images/' # img实际存放地址
Annotation_path = '/home/lc/data_center/gift/v2/xml/' # xml实际存放地址
label_path = '/home/lc/data_center/gift/v2/labels/' # 保存路径
# wd = getcwd()
imgname_list = []