From 182009f4e4b63fdc83b43e266a8fb2b716e9cbbf Mon Sep 17 00:00:00 2001
From: Bubbliiiing <47347516+bubbliiiing@users.noreply.github.com>
Date: Tue, 8 Sep 2020 12:46:08 +0800
Subject: [PATCH] Update get_dr_txt.py

---
 get_dr_txt.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/get_dr_txt.py b/get_dr_txt.py
index 82dff30..f81c08b 100644
--- a/get_dr_txt.py
+++ b/get_dr_txt.py
@@ -15,7 +15,7 @@ from yolo import YOLO
 from nets.yolo4 import YoloBody
 from PIL import Image,ImageFont, ImageDraw
 from utils.utils import non_max_suppression, bbox_iou, DecodeBox,letterbox_image,yolo_correct_boxes
-
+from tqdm import tqdm
 class mAP_Yolo(YOLO):
     #---------------------------------------------------#
     #   检测图片
@@ -83,13 +83,11 @@ if not os.path.exists("./input/images-optional"):
     os.makedirs("./input/images-optional")
 
 
-for image_id in image_ids:
+for image_id in tqdm(image_ids):
     image_path = "./VOCdevkit/VOC2007/JPEGImages/"+image_id+".jpg"
     image = Image.open(image_path)
     # 开启后在之后计算mAP可以可视化
     # image.save("./input/images-optional/"+image_id+".jpg")
     yolo.detect_image(image_id,image)
-    print(image_id," done!")
     
-
 print("Conversion completed!")
-- 
GitLab