Skip to content
Snippets Groups Projects
Commit 7159fd15 authored by Stephane Chavin's avatar Stephane Chavin
Browse files

correction

parent be416a22
No related branches found
No related tags found
No related merge requests found
......@@ -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')
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment