Export original VOC format in labelme2voc.py

examples/instance_segmentation/README.md

@@ -15,8 +15,10 @@ labelme data_annotated --labels labels.txt --nodata --labelflags '{.*: [occluded
 # It generates:
 #   - data_dataset_voc/JPEGImages
 #   - data_dataset_voc/SegmentationClass
+#   - data_dataset_voc/SegmentationClassNpy
 #   - data_dataset_voc/SegmentationClassVisualization
 #   - data_dataset_voc/SegmentationObject
+#   - data_dataset_voc/SegmentationObjectNpy
 #   - data_dataset_voc/SegmentationObjectVisualization
 ./labelme2voc.py data_annotated data_dataset_voc --labels labels.txt
 ```
@@ -30,8 +32,8 @@ Note that the label file contains only very low label values (ex. `0, 4, 14`), a
 You can see the label PNG file by following.
 
 ```bash
-labelme_draw_label_png data_dataset_voc/SegmentationClassPNG/2011_000003.png   # left
-labelme_draw_label_png data_dataset_voc/SegmentationObjectPNG/2011_000003.png  # right
+labelme_draw_label_png data_dataset_voc/SegmentationClass/2011_000003.png   # left
+labelme_draw_label_png data_dataset_voc/SegmentationObject/2011_000003.png  # right
 ```
 
 <img src=".readme/draw_label_png_class.jpg" width="33%" /> <img src=".readme/draw_label_png_object.jpg" width="33%" />

examples/instance_segmentation/labelme2voc.py

@@ -18,11 +18,17 @@ def main():
     parser = argparse.ArgumentParser(
         formatter_class=argparse.ArgumentDefaultsHelpFormatter
     )
-    parser.add_argument("input_dir", help="input annotated directory")
-    parser.add_argument("output_dir", help="output dataset directory")
-    parser.add_argument("--labels", help="labels file", required=True)
+    parser.add_argument("input_dir", help="Input annotated directory")
+    parser.add_argument("output_dir", help="Output dataset directory")
+    parser.add_argument("--labels", help="Labels file", required=True)
     parser.add_argument(
-        "--noviz", help="no visualization", action="store_true"
+        "--noobject", help="Flag not to generate object label", action="store_true"
+    )
+    parser.add_argument(
+        "--nonpy", help="Flag not to generate .npy files", action="store_true"
+    )
+    parser.add_argument(
+        "--noviz", help="Flag to disable visualization", action="store_true"
     )
     args = parser.parse_args()
 
@@ -32,17 +38,16 @@ def main():
     os.makedirs(args.output_dir)
     os.makedirs(osp.join(args.output_dir, "JPEGImages"))
     os.makedirs(osp.join(args.output_dir, "SegmentationClass"))
-    os.makedirs(osp.join(args.output_dir, "SegmentationClassPNG"))
-    if not args.noviz:
-        os.makedirs(
-            osp.join(args.output_dir, "SegmentationClassVisualization")
-        )
-    os.makedirs(osp.join(args.output_dir, "SegmentationObject"))
-    os.makedirs(osp.join(args.output_dir, "SegmentationObjectPNG"))
+    if not args.nonpy:
+        os.makedirs(osp.join(args.output_dir, "SegmentationClassNpy"))
     if not args.noviz:
-        os.makedirs(
-            osp.join(args.output_dir, "SegmentationObjectVisualization")
-        )
+        os.makedirs(osp.join(args.output_dir, "SegmentationClassVisualization"))
+    if not args.noobject:
+        os.makedirs(osp.join(args.output_dir, "SegmentationObject"))
+        if not args.nonpy:
+            os.makedirs(osp.join(args.output_dir, "SegmentationObjectNpy"))
+        if not args.noviz:
+            os.makedirs(osp.join(args.output_dir, "SegmentationObjectVisualization"))
     print("Creating dataset:", args.output_dir)
 
     class_names = []
@@ -71,30 +76,31 @@ def main():
 
         base = osp.splitext(osp.basename(filename))[0]
         out_img_file = osp.join(args.output_dir, "JPEGImages", base + ".jpg")
-        out_cls_file = osp.join(
-            args.output_dir, "SegmentationClass", base + ".npy"
-        )
-        out_clsp_file = osp.join(
-            args.output_dir, "SegmentationClassPNG", base + ".png"
-        )
+        out_clsp_file = osp.join(args.output_dir, "SegmentationClass", base + ".png")
+        if not args.nonpy:
+            out_cls_file = osp.join(
+                args.output_dir, "SegmentationClassNpy", base + ".npy"
+            )
         if not args.noviz:
             out_clsv_file = osp.join(
                 args.output_dir,
                 "SegmentationClassVisualization",
                 base + ".jpg",
             )
-        out_ins_file = osp.join(
-            args.output_dir, "SegmentationObject", base + ".npy"
-        )
-        out_insp_file = osp.join(
-            args.output_dir, "SegmentationObjectPNG", base + ".png"
-        )
-        if not args.noviz:
-            out_insv_file = osp.join(
-                args.output_dir,
-                "SegmentationObjectVisualization",
-                base + ".jpg",
+        if not args.noobject:
+            out_insp_file = osp.join(
+                args.output_dir, "SegmentationObject", base + ".png"
             )
+            if not args.nonpy:
+                out_ins_file = osp.join(
+                    args.output_dir, "SegmentationObjectNpy", base + ".npy"
+                )
+            if not args.noviz:
+                out_insv_file = osp.join(
+                    args.output_dir,
+                    "SegmentationObjectVisualization",
+                    base + ".jpg",
+                )
 
         img = labelme.utils.img_data_to_arr(label_file.imageData)
         imgviz.io.imsave(out_img_file, img)
@@ -108,7 +114,8 @@ def main():
 
         # class label
         labelme.utils.lblsave(out_clsp_file, cls)
-        np.save(out_cls_file, cls)
+        if not args.nonpy:
+            np.save(out_cls_file, cls)
         if not args.noviz:
             clsv = imgviz.label2rgb(
                 cls,
@@ -119,20 +126,22 @@ def main():
             )
             imgviz.io.imsave(out_clsv_file, clsv)
 
-        # instance label
-        labelme.utils.lblsave(out_insp_file, ins)
-        np.save(out_ins_file, ins)
-        if not args.noviz:
-            instance_ids = np.unique(ins)
-            instance_names = [str(i) for i in range(max(instance_ids) + 1)]
-            insv = imgviz.label2rgb(
-                ins,
-                imgviz.rgb2gray(img),
-                label_names=instance_names,
-                font_size=15,
-                loc="rb",
-            )
-            imgviz.io.imsave(out_insv_file, insv)
+        if not args.noobject:
+            # instance label
+            labelme.utils.lblsave(out_insp_file, ins)
+            if not args.nonpy:
+                np.save(out_ins_file, ins)
+            if not args.noviz:
+                instance_ids = np.unique(ins)
+                instance_names = [str(i) for i in range(max(instance_ids) + 1)]
+                insv = imgviz.label2rgb(
+                    ins,
+                    imgviz.rgb2gray(img),
+                    label_names=instance_names,
+                    font_size=15,
+                    loc="rb",
+                )
+                imgviz.io.imsave(out_insv_file, insv)
 
 
 if __name__ == "__main__":

examples/semantic_segmentation/README.md

@@ -15,11 +15,12 @@ labelme data_annotated --labels labels.txt --nodata --validatelabel exact --conf
 # It generates:
 #   - data_dataset_voc/JPEGImages
 #   - data_dataset_voc/SegmentationClass
+#   - data_dataset_voc/SegmentationClassNpy
 #   - data_dataset_voc/SegmentationClassVisualization
-./labelme2voc.py data_annotated data_dataset_voc --labels labels.txt
+./labelme2voc.py data_annotated data_dataset_voc --labels labels.txt --noobject
 ```
 
-<img src="data_dataset_voc/JPEGImages/2011_000003.jpg" width="33%" /> <img src="data_dataset_voc/SegmentationClassPNG/2011_000003.png" width="33%" /> <img src="data_dataset_voc/SegmentationClassVisualization/2011_000003.jpg" width="33%" />
+<img src="data_dataset_voc/JPEGImages/2011_000003.jpg" width="33%" /> <img src="data_dataset_voc/SegmentationClass/2011_000003.png" width="33%" /> <img src="data_dataset_voc/SegmentationClassVisualization/2011_000003.jpg" width="33%" />
 
 Fig 1. JPEG image (left), PNG label (center), JPEG label visualization (right)  
 
@@ -29,7 +30,7 @@ Note that the label file contains only very low label values (ex. `0, 4, 14`), a
 You can see the label PNG file by following.
 
 ```bash
-labelme_draw_label_png data_dataset_voc/SegmentationClassPNG/2011_000003.png
+labelme_draw_label_png data_dataset_voc/SegmentationClass/2011_000003.png
 ```
 
 <img src=".readme/draw_label_png.jpg" width="33%" />

examples/semantic_segmentation/labelme2voc.py

@@ -0,0 +1 @@
+../instance_segmentation/labelme2voc.py