Updated json files to COCO format, updated the height and width for s…

…ome images
TempleRAIL · Jul 16, 2021 · cb8079c · cb8079c
1 parent 00dae2d
commit cb8079c
Show file tree

Hide file tree

Showing 7 changed files with 45 additions and 41 deletions.
diff --git a/cnn/image_size.py b/cnn/image_size.py
diff --git a/cnn/convert_json.py → cnn/json_tools/convert_json.py b/cnn/convert_json.py → cnn/json_tools/convert_json.py
diff --git a/cnn/json_tools/image_size.py b/cnn/json_tools/image_size.py
@@ -0,0 +1,44 @@
+import os
+from PIL import Image
+from pathlib import Path
+import json
+
+img_dir = './sherd_datasets/validation/'
+
+# opening json file in read mode
+json_file = open('./sherd_datasets/validation/val_annotations_coco.json', 'r')
+
+# converting json file to python dictionary
+data = json.load(json_file)
+
+# closing json file
+json_file.close()
+print('json_file closed\n')
+
+# this loop iterates the jpg files
+pathlist = Path(img_dir).glob('*.jpg')
+for path in pathlist:
+    path_str = str(path)
+    img = Image.open(path)
+    for i in data['images']:
+        if i['file_name'] == path_str[24:]:
+            i['width'] = img.size[0]
+            i['height'] = img.size[1]
+
+# this loop iterates the png files
+pathlist = Path(img_dir).glob('*.png')
+for path in pathlist:
+    path_str = str(path)
+    img = Image.open(path)
+    for i in data['images']:
+        if i['file_name'] == path_str[24:]:
+            i['width'] = img.size[0]
+            i['height'] = img.size[1]
+
+# opening json file in write mode and overwriting data
+json_file = open('./sherd_datasets/validation/val_annotations_coco.json', 'w')
+print('\nvia.json file opened\n')
+json.dump(data, json_file)
+print('via.json file overwritten\n')
+json_file.close()
+print('via.json file closed\n')
diff --git a/cnn/read_json.py → cnn/json_tools/read_json.py b/cnn/read_json.py → cnn/json_tools/read_json.py
diff --git a/cnn/string_to_int.py → cnn/json_tools/string_to_int.py b/cnn/string_to_int.py → cnn/json_tools/string_to_int.py
diff --git a/cnn/update_list.py → cnn/json_tools/update_list.py b/cnn/update_list.py → cnn/json_tools/update_list.py
diff --git a/cnn/sherd_datasets/training/train_annotations_coco.json b/cnn/sherd_datasets/training/train_annotations_coco.json