From f4c4e589bc693284eaa6b6ede74951569bb61075 Mon Sep 17 00:00:00 2001
From: Bubbliiiing <47347516+bubbliiiing@users.noreply.github.com>
Date: Thu, 22 Oct 2020 10:35:08 +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 ae354a2..1235ef3 100644
--- a/get_dr_txt.py
+++ b/get_dr_txt.py
@@ -22,6 +22,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