diff --git a/get_dr_txt.py b/get_dr_txt.py index 7b59761..40140c9 100644 --- a/get_dr_txt.py +++ b/get_dr_txt.py @@ -16,13 +16,14 @@ from PIL import Image,ImageFont, ImageDraw from utils.config import Config from utils.utils import non_max_suppression, bbox_iou, DecodeBox,letterbox_image,yolo_correct_boxes +from tqdm import tqdm class mAP_Yolo(YOLO): #---------------------------------------------------# # 检测图片 #---------------------------------------------------# def detect_image(self,image_id,image): - self.confidence = 0.05 + self.confidence = 0.001 f = open("./input/detection-results/"+image_id+".txt","w") image_shape = np.array(np.shape(image)[0:2]) @@ -82,13 +83,11 @@ def detect_image(self,image_id,image): 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!")