From bc720d7e8effd4d676caed8ab648fa6ff8099724 Mon Sep 17 00:00:00 2001
From: Bubbliiiing <47347516+bubbliiiing@users.noreply.github.com>
Date: Fri, 11 Sep 2020 14:09:38 +0800
Subject: [PATCH] Update get_dr_txt.py

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

diff --git a/get_dr_txt.py b/get_dr_txt.py
index f81c08b..46823c2 100644
--- a/get_dr_txt.py
+++ b/get_dr_txt.py
@@ -46,7 +46,7 @@ class mAP_Yolo(YOLO):
         output = torch.cat(output_list, 1)
         batch_detections = non_max_suppression(output, len(self.class_names),
                                                 conf_thres=self.confidence,
-                                                nms_thres=0.3)
+                                                nms_thres=self.iou)
 
         try:
             batch_detections = batch_detections[0].cpu().numpy()
-- 
GitLab