Update get_gt_txt.py

get_gt_txt.py

+#----------------------------------------------------#
+#   获取测试集的ground-truth
+#   具体视频教程可查看
+#   https://www.bilibili.com/video/BV1zE411u7Vw
+#----------------------------------------------------#
 import sys
 import os
 import glob
@@ -14,6 +19,10 @@ for image_id in image_ids:
     with open("./input/ground-truth/"+image_id+".txt", "w") as new_f:
         root = ET.parse("VOCdevkit/VOC2007/Annotations/"+image_id+".xml").getroot()
         for obj in root.findall('object'):
+            if obj.find('difficult')!=None:
+                difficult = obj.find('difficult').text
+                if int(difficult)==1:
+                    continue
             obj_name = obj.find('name').text
             bndbox = obj.find('bndbox')
             left = bndbox.find('xmin').text
@@ -21,4 +30,5 @@ for image_id in image_ids:
             right = bndbox.find('xmax').text
             bottom = bndbox.find('ymax').text
             new_f.write("%s %s %s %s %s\n" % (obj_name, left, top, right, bottom))
-print("Conversion completed!")
\ No newline at end of file
+            
+print("Conversion completed!")