ekurtic commited on
Commit
e7888af
·
unverified ·
1 Parent(s): 8c6f9e1

Fix inconsistent NMS IoU value for COCO (#3934)

Browse files

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

Files changed (1) hide show
  1. train.py +0 -2
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,