From 6298c177bb08490271c9734d874cf362ba442332 Mon Sep 17 00:00:00 2001
From: raphael <raphael@maenle.net>
Date: Mon, 16 Aug 2021 16:32:21 +0200
Subject: [PATCH] adds generator script for cvat data

---
 coco_cvat_gen.py | 72 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 72 insertions(+)
 create mode 100644 coco_cvat_gen.py

diff --git a/coco_cvat_gen.py b/coco_cvat_gen.py
new file mode 100644
index 0000000..ccc867e
--- /dev/null
+++ b/coco_cvat_gen.py
@@ -0,0 +1,72 @@
+import json, os
+import matplotlib.pyplot as plt
+import numpy as np
+
+from PIL import Image
+
+no_label = 0
+small = 0
+passed = 0
+count = 0
+
+
+
+print("loading coco annotations...")
+coco = json.load(open('./coco/annotations/instances_default.json'))
+print("done")
+
+def findAnnotationName(annotationId):
+    for c in coco['categories']:
+        if c['id'] == annotationId:
+            return c['name']
+
+def findAnnotationToId(ident):
+    for annotation in coco['annotations']:
+        img_an = annotation['image_id']
+        if img_an == ident:
+            return annotation 
+
+def show(pil, pause=0.2):
+    ImageNumpyFormat = np.asarray(pil)
+    plt.imshow(ImageNumpyFormat)
+    plt.draw()
+    plt.pause(pause) # pause how many seconds
+    plt.close()
+
+
+def parseImage(coImg):
+    global no_label, small, passed, coco
+    # open image file
+
+    path = "coco/roto_frank/images/" + coImg['file_name'].split('/')[4]
+    img = Image.open(path)
+
+    an = findAnnotationToId(coImg['id'])
+    if an == None:
+        no_label += 1
+        return
+
+    c = an['bbox']
+    crop = img.crop((c[0], c[1], c[0]+c[2], c[1]+c[3]))
+
+    if crop.width < 64 or crop.height < 64:
+        small += 1
+        return
+
+    imagePath = f"classified_roto/{findAnnotationName(an['category_id'])}/{an['id']}.png"
+    os.makedirs(os.path.dirname(imagePath), exist_ok=True)
+    crop.save(imagePath)
+    passed += 1
+
+if __name__ == "__main__":
+
+    for coImg in coco['images']:
+        parseImage(coImg)
+        count += 1
+        if count % 100 == 0:
+            print("status:")
+            print(f"no labels: {no_label}")
+            print(f"to small: {small}")
+            print(f"passed: {passed}")
+            print("-----")
+