diff --git a/get_dr_txt.py b/get_dr_txt.py
index f81c08bfc1369fad189775b0372aa37036cb99c6..46823c23a58d8a2c763b5b415f8269c2e5f7ed0d 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()