| @@ -36,6 +36,8 @@ class COCO2YOLO: | |||
| for image in self.labels['images']: | |||
| id = image['id'] | |||
| file_name = image['file_name'] | |||
| if file_name.find('\\') > -1: | |||
| file_name = file_name[file_name.index('\\')+1:] | |||
| w = image['width'] | |||
| h = image['height'] | |||
| images_info[id] = (file_name, w, h) | |||