first commit

DetResizeForTrain.py

@@ -0,0 +1,37 @@
+import os
+import cv2
+import shutil
+from tqdm import tqdm
+anno_pathRoot = "/data/DataManagement/Project/CAS_ProjectBird/CDB_ProjectBird/Annotations_Original/"
+image_pathRoot = anno_pathRoot.replace("/Annotations_Original/","/JPEGImages_Original/")
+w = 1024
+h = 768
+
+def DetResizeForTrain(anno_pathRoot,image_pathRoot,w,h):
+    annoResultPath = anno_pathRoot.replace("/Annotations_Original/","/Annotations/")
+    annoJPEGPath = anno_pathRoot.replace("/JPEGImages_Original/", "/JPEGImages/")
+    for anno in tqdm(os.listdir(anno_pathRoot)):
+        anno = GEN_Annotations(anno.replace(".xml", ".jpg"))
+        dom = xml.dom.minidom.parse(os.path.join(anno_pathRoot, xmls))
+        size = dom.getElementsByTagName("size")
+        # print(size[0].getElementsByTagName('width')[0].childNodes[0].data)
+        width = size[0].getElementsByTagName('width')[0].childNodes[0].data
+        height = size[0].getElementsByTagName('height')[0].childNodes[0].data
+        depth = size[0].getElementsByTagName('depth')[0].childNodes[0].data
+        anno.set_size(int(w), int(h), int(depth))
+        objects = dom.getElementsByTagName("object")
+        for object in objects:
+            bndbox = object.getElementsByTagName('bndbox')[0]
+            xmin = bndbox.getElementsByTagName('xmin')[0]
+            ymin = bndbox.getElementsByTagName('ymin')[0]
+            xmax = bndbox.getElementsByTagName('xmax')[0]
+            ymax = bndbox.getElementsByTagName('ymax')[0]
+
+            xmin_data = int(xmin.childNodes[0].data)*(w/width)
+            ymin_data = int(ymin.childNodes[0].data)*(h/height)
+            xmax_data = int(xmax.childNodes[0].data)*(w/width)
+            ymax_data = int(ymax.childNodes[0].data)*(h/height)
+            label_name = object.getElementsByTagName('name')[0].childNodes[0].data
+            anno.add_pic_attr(label=label_name, xmin=xmin_data, xmax=xmax_data, ymin=ymin_data, ymax=ymax_data)
+        anno.savefile(os.path.join(annoResultPath, xmls))
+        image = cv2.imread