mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity

Fix inconsistent NMS IoU value for COCO (#3934)

Browse Source 
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
This commit is contained in:
Eldar Kurtic 2021-07-08 15:29:02 +02:00 committed by GitHub
parent 8c6f9e15bf
commit e7888af94c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

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