Add raise when class is blank or space between words (#391)

pull/393/head
HinGwenWoong 2022-12-19 16:55:33 +08:00 committed by GitHub
parent e0128c2dbd
commit 446dfbd4b9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 1 deletions

View File

@ -274,7 +274,15 @@ def convert_labelme_to_coco(image_dir: str,
assert len(txt_lines) > 0
for txt_line in txt_lines:
v, k = txt_line.split(' ')
class_info = txt_line.split(' ')
if len(class_info) != 2:
raise ValueError('Error parse "class_id_txt" file '
f'{class_id_txt}, please check if some of '
'the class names is blank, like "1 " -> '
'"1 blank", or class name has space between'
' words, like "1 Big house" -> "1 '
'Big-house".')
v, k = class_info
all_classes_id.update({k: v})
else:
all_classes_id = None