diff --git a/labelme2yolo.py b/labelme2yolo.py index f526962f5a079445ca9820a4280532fc29844883..6c1a7c0955ee4a659331a45d217bf81beed4719a 100644 --- a/labelme2yolo.py +++ b/labelme2yolo.py @@ -30,10 +30,10 @@ for index, labelme_annotation_path in enumerate(glob(f'{args.path_to_data}/*.jso labelme_annotation_file = open(labelme_annotation_path, 'r') labelme_annotation = json.load(labelme_annotation_file) - yolo_annotation_path = os.path.join(args.direction,'labels', image_id + '.txt') + yolo_annotation_path = os.path.join(args.directory,'labels', image_id + '.txt') yolo_annotation_file = open(yolo_annotation_path, 'w') yolo_image = base64.decodebytes(labelme_annotation['imageData'].encode()) - yolo_image_path = os.path.join(args.direction, 'labels', image_id + '.jpg') + yolo_image_path = os.path.join(args.directory, 'labels', image_id + '.jpg') # Write YOLO image (and it to the list) yolo_image_file = open(yolo_image_path, 'wb')