Unverified 提交 e7888af9 authored 作者: Eldar Kurtic's avatar Eldar Kurtic 提交者: GitHub

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
上级 8c6f9e15
...@@ -457,8 +457,6 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary ...@@ -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,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论