Update get_gt_txt.py

wyf763240785 · May 20, 2020 · 63db2bd · 63db2bd
1 parent 9c4c71c
commit 63db2bd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/get_gt_txt.py b/get_gt_txt.py
@@ -16,7 +16,7 @@
     os.makedirs("./input/ground-truth")
 
 for image_id in image_ids:
-    with open("./input/ground-truth/"+image_id+".txt", "a") as new_f:
+    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'):
             obj_name = obj.find('name').text
@@ -26,4 +26,4 @@
             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!")
+print("Conversion completed!")