Browse Source

Create testing.py

LaraStuStu 5 years ago
parent
commit
8d11958b46
1 changed files with 25 additions and 0 deletions
  1. 25 0
      DataAnnotation/labelme/labelme/testing.py

+ 25 - 0
DataAnnotation/labelme/labelme/testing.py

@@ -0,0 +1,25 @@
+import json
+import os.path as osp
+
+import labelme.utils
+
+
+def assert_labelfile_sanity(filename):
+    assert osp.exists(filename)
+
+    data = json.load(open(filename))
+
+    assert 'imagePath' in data
+    imageData = data.get('imageData', None)
+    if imageData is None:
+        assert osp.exists(data['imagePath'])
+    img = labelme.utils.img_b64_to_arr(imageData)
+
+    H, W = img.shape[:2]
+    assert 'shapes' in data
+    for shape in data['shapes']:
+        assert 'label' in shape
+        assert 'points' in shape
+        for x, y in shape['points']:
+            assert 0 <= x <= W
+            assert 0 <= y <= H