first

detection_negative_samples.py

@@ -1,2 +1,88 @@
 import os
 import cv2
+import random
+from PIL import Image
+from lxml import etree
+path_unit= "/data1/zwq/model/data_MPKFNet/data/ConstructionSite/Hatcher_One"
+path_raw = "/data1/zwq/model/data_MPKFNet/data/ConstructionSite/NO"
+result_path = "/data1/zwq/model/data_MPKFNet/data/ConstructionSite/result"
+
+class GEN_Annotations:
+    def __init__(self, filename):
+        self.root = etree.Element("annotation")
+
+        child1 = etree.SubElement(self.root, "folder")
+        child1.text = "VOC2007"
+
+        child2 = etree.SubElement(self.root, "filename")
+        child2.text = filename
+
+        child3 = etree.SubElement(self.root, "source")
+
+        child4 = etree.SubElement(child3, "annotation")
+        child4.text = "PASCAL VOC2007"
+        child5 = etree.SubElement(child3, "database")
+        child5.text = "Unknown"
+
+        child6 = etree.SubElement(child3, "image")
+        child6.text = "flickr"
+        child7 = etree.SubElement(child3, "flickrid")
+        child7.text = "35435"
+
+    def set_size(self, witdh, height, channel):
+        size = etree.SubElement(self.root, "size")
+        widthn = etree.SubElement(size, "width")
+        widthn.text = str(witdh)
+        heightn = etree.SubElement(size, "height")
+        heightn.text = str(height)
+        channeln = etree.SubElement(size, "depth")
+        channeln.text = str(channel)
+
+    def savefile(self, filename):
+        tree = etree.ElementTree(self.root)
+        tree.write(filename, pretty_print=True, xml_declaration=False, encoding='utf-8')
+
+    def add_pic_attr(self, label, xmin, ymin, xmax, ymax):
+        object = etree.SubElement(self.root, "object")
+        namen = etree.SubElement(object, "name")
+        namen.text = label
+        bndbox = etree.SubElement(object, "bndbox")
+        xminn = etree.SubElement(bndbox, "xmin")
+        xminn.text = str(xmin)
+        yminn = etree.SubElement(bndbox, "ymin")
+        yminn.text = str(ymin)
+        xmaxn = etree.SubElement(bndbox, "xmax")
+        xmaxn.text = str(xmax)
+        ymaxn = etree.SubElement(bndbox, "ymax")
+        ymaxn.text = str(ymax)
+
+
+def negative_samples():
+    unit_list = os.listdir(path_unit)
+    unit_img_list = []
+    for unit_img in unit_list:
+        img1 = Image.open(os.path.join(path_unit,unit_img))
+        unit_img_list.append(img1)
+    raw_list = os.listdir(path_raw)
+    for raw_path  in raw_list:
+        print(raw_list)
+        print("\n")
+        raw_img = Image.open(os.path.join(path_raw,raw_path))
+        unit_num = random.randint(0,10)
+        x_left = random.randint(0,raw_img.size[0]-unit_img_list[unit_num].size[0])
+        y_left = random.randint(0,raw_img.size[1]-unit_img_list[unit_num].size[1])
+        x_right = x_left+unit_img_list[unit_num].size[0]
+        y_right = y_left+unit_img_list[unit_num].size[1]
+        raw_img.paste(unit_img_list[unit_num],(x_left,y_left,x_right,y_right))
+        raw_img.save(os.path.join(path_raw,raw_path).replace("/NO/","/result/Images/"))
+        filename = raw_path
+        anno = GEN_Annotations(filename)
+        anno.set_size(raw_img.size[0], raw_img.size[1], 3)
+        anno.add_pic_attr("person", x_left, y_left, x_right, y_right)
+        anno.savefile(os.path.join(path_raw,raw_path).replace("/NO/","/result/Ground_truth/").replace(".jpg",".xml"))
+
+
+
+if __name__ == '__main__':
+
+    negative_samples()