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

Merge branch 'master' of https://gitee.com/nanjing-yimao-information/ieemoo-ai-gift

Browse Source
This commit is contained in:
lichen
2025-06-21 13:42:44 +08:00
parent b0946fab2d 201d0bccc0
commit a17373d576
6 changed files with 2174 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
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
# 标注越界修正