Fix inconsistent NMS IoU value for COCO (#3934)
Browse filesEvaluation 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
CHANGED
@@ -457,8 +457,6 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary
|
|
457 |
results, _, _ = test.run(data_dict,
|
458 |
batch_size=batch_size // WORLD_SIZE * 2,
|
459 |
imgsz=imgsz_test,
|
460 |
-
conf_thres=0.001,
|
461 |
-
iou_thres=0.7,
|
462 |
model=attempt_load(m, device).half(),
|
463 |
single_cls=single_cls,
|
464 |
dataloader=testloader,
|
|
|
457 |
results, _, _ = test.run(data_dict,
|
458 |
batch_size=batch_size // WORLD_SIZE * 2,
|
459 |
imgsz=imgsz_test,
|
|
|
|
|
460 |
model=attempt_load(m, device).half(),
|
461 |
single_cls=single_cls,
|
462 |
dataloader=testloader,
|