From 452e84703c0a0afe1f7e04bc9f2369ba77b2a9fb Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 30 Sep 2021 13:57:53 -0700 Subject: [PATCH] Update train.py (#5014) --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 39fe1a0cb..878c9a07f 100644 --- a/train.py +++ b/train.py @@ -412,7 +412,7 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary batch_size=batch_size // WORLD_SIZE * 2, imgsz=imgsz, model=attempt_load(f, device).half(), - iou_thres=0.7 if is_coco else 0.6, # best pycocotools results at 0.7 + iou_thres=0.65 if is_coco else 0.60, # best pycocotools results at 0.65 single_cls=single_cls, dataloader=val_loader, save_dir=save_dir,