diff --git a/train.py b/train.py index 39fe1a0..878c9a0 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,