python: introduce markers and crop scripts to help debug

python/crop.py

@@ -0,0 +1,26 @@
+import os
+import argparse
+import imtool
+
+parser = argparse.ArgumentParser(description='crop images to train YOLO on squares')
+parser.add_argument('src', metavar='dir', type=str, nargs='+',
+                    help='dir containing the images')
+parser.add_argument('--dst', dest='dst', type=str, default='./data/squares',
+                    help='dest dir')
+
+args = parser.parse_args()
+
+for d in args.src:
+    i = 0
+    with os.scandir(d) as it:
+        for e in it:
+            if e.name.endswith('.png') and e.is_file():
+                print(e.name)
+                label = e.path.replace('images', 'labels').replace('.png', '.txt')
+                try:
+                    i+=1
+                    bco, boxes = imtool.read_centroids(label)
+                    imtool.crop(bco, e.path, boxes, args.dst)
+
+                except Exception as err:
+                    print(err)

python/markers.py

@@ -0,0 +1,21 @@
+import cv2
+import argparse
+import imtool
+
+parser = argparse.ArgumentParser(description='shows YOLO markers')
+parser.add_argument('pngs', metavar='img.png', type=str, nargs='+',
+                    help='images to debug')
+args = parser.parse_args()
+
+for i in args.pngs:
+    im = cv2.imread(i)
+    label = i.replace('images', 'labels').replace('.png', '.txt').replace('.jpg', '.txt')
+    bco, ccs = imtool.read_centroids(label)
+    bbs = [c.to_bounding_box(im.shape) for c in ccs]
+    for i,b in enumerate(bbs):
+        c = (100, 255*i/len(bbs), 255*(1 - i/len(bbs)))
+        cv2.rectangle(im, b.start, b.end, c, 5)
+
+cv2.imshow('result', im)
+cv2.waitKey(0)
+cv2.destroyAllWindows()