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 ultralytics/yolov5#3907
SecretStar112 · Jul 8, 2021 · 828cda1 · 828cda1
1 parent e893a91
commit 828cda1
Showing 1 changed file with 0 additions and 2 deletions.
diff --git 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,