first commit

detectionCorrectxml.py

@@ -0,0 +1,40 @@
+import os
+import cv2
+import  xml.dom.minidom
+from tqdm import tqdm
+from lxml import etree
+from creat_xml import GEN_Annotations
+xml_path = "/data/DataManagement/Project/CAS_ProjectBird/CDB_ProjectBird/Annotations"
+image_path = xml_path.replace("/Annotations","/JPEGImages")
+for xmls in tqdm(os.listdir(xml_path)):
+    anno = GEN_Annotations(xmls.replace(".xml",".jpg"))
+    dom = xml.dom.minidom.parse(os.path.join(xml_path,xmls))
+    try:
+        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
+    except:
+
+        print(xmls)
+        image = cv2.imread(os.path.join(image_path,xmls.replace(".xml",".jpg")))
+        width = image.shape[1]
+        height = image.shape[0]
+        depth = image.shape[2]
+    anno.set_size(int(width), int(height), 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)
+        ymin_data = int(ymin.childNodes[0].data)
+        xmax_data = int(xmax.childNodes[0].data)
+        ymax_data = int(ymax.childNodes[0].data)
+        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(xml_path,xmls).replace("/Annotations/","/Annotations_temp/"))