correction labelme x min

labelme2yolo.py

@@ -12,6 +12,15 @@ def arg_directory(path):
     else:
         raise argparse.ArgumentTypeError(f'`{path}` is not a valid path')
 
+def create_directory(directory):
+    try:
+        os.makedirs(os.path.join(directory,'labels'), exist_ok=True)
+    except Exception as e:
+        print(f'Error creating directory {directory}: {e}')
+
+    return (directory)
+
+
 def convert_labelme_to_yolo(labelme_annotation_path, yolo_directory):
     # Load LabelMe annotation
     image_id = Path(labelme_annotation_path).stem
@@ -36,14 +45,18 @@ def convert_labelme_to_yolo(labelme_annotation_path, yolo_directory):
                 continue
 
             label = shape['label']
-            x1, y1 = shape['points'][0]
-            x2, y2 = shape['points'][1]
-            width = x2 - x1
-            height = y2 - y1
-            x_center = (x1 + x2) / 2
-            y_center = (y1 + y2) / 2
-
-            annotation_line = f'{label} {x_center} {y_center} {width} {height}\n'
+
+            # shape['points'] format : [[x1,y1],[x2,y2]...] #
+            scale_width = 1.0 / labelme_annotation['imageWidth']
+            scale_height = 1.0 / labelme_annotation['imageHeight']
+            width = abs(shape['points'][1][0] - shape['points'][0][0]) * scale_width
+            height = abs(shape['points'][1][1] - shape['points'][0][1]) * scale_height
+
+            x = min(shape['points'][0][0], shape['points'][1][0]) * scale_width + width / 2
+            y = min(shape['points'][0][1], shape['points'][1][1]) * scale_height + height / 2
+            if x+width/2 > 1 or y+height/2>1:
+                print(f'Error with bounding box values over 1 in file {yolo_image_file}')
+            annotation_line = f'{label} {x} {y} {width} {height}\n'
             yolo_annotation_file.write(annotation_line)
 
 def main(args):