From 528031cec65280ea4c3b8ca3fcae6d06f02cc148 Mon Sep 17 00:00:00 2001
From: Bubbliiiing <47347516+bubbliiiing@users.noreply.github.com>
Date: Thu, 22 Oct 2020 10:05:07 +0800
Subject: [PATCH] Update get_dr_txt.py

---
 get_dr_txt.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/get_dr_txt.py b/get_dr_txt.py
index 6f82dcc..9c0547b 100644
--- a/get_dr_txt.py
+++ b/get_dr_txt.py
@@ -24,6 +24,7 @@ class mAP_Yolo(YOLO):
     #---------------------------------------------------#
     def detect_image(self,image_id,image):
         self.confidence = 0.01
+        self.iou = 0.5
         f = open("./input/detection-results/"+image_id+".txt","w") 
         image_shape = np.array(np.shape(image)[0:2])
 
-- 
GitLab