diff --git a/get_gt_txt.py b/get_gt_txt.py index 4acfa2844b17899ada6716ce0403c97b76c3b664..c5d9b0895cf4798ac5a28b2e713ff0342daa09d5 100644 --- a/get_gt_txt.py +++ b/get_gt_txt.py @@ -7,6 +7,7 @@ import sys import os import glob import xml.etree.ElementTree as ET +from tqdm import tqdm image_ids = open('VOCdevkit/VOC2007/ImageSets/Main/test.txt').read().strip().split() @@ -15,7 +16,7 @@ if not os.path.exists("./input"): if not os.path.exists("./input/ground-truth"): os.makedirs("./input/ground-truth") -for image_id in image_ids: +for image_id in tqdm(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'):