From 828cda1518d50d38325600a94d3166846cb37480 Mon Sep 17 00:00:00 2001
From: SecretStar112 <edwardxian062@gmail.com>
Date: Thu, 8 Jul 2021 15:29:02 +0200
Subject: [PATCH] Fix inconsistent NMS IoU value for COCO (#3934)

Evaluation of 'best' and 'last' models will use the same params as the evaluation during the training phase.
This PR fixes https://github.com/ultralytics/yolov5/issues/3907
---
 train.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/train.py b/train.py
index 5a43477..e58d7c4 100644
--- a/train.py
+++ b/train.py
@@ -457,8 +457,6 @@ def train(hyp,  # path/to/hyp.yaml or hyp dictionary
                     results, _, _ = test.run(data_dict,
                                              batch_size=batch_size // WORLD_SIZE * 2,
                                              imgsz=imgsz_test,
-                                             conf_thres=0.001,
-                                             iou_thres=0.7,
                                              model=attempt_load(m, device).half(),
                                              single_cls=single_cls,
                                              dataloader=testloader,