Update train.py (#5014)

2021-09-30 13:57:53 -07:00 · 2021-09-30 13:57:53 -07:00 · 452e84703c
parent cce313d67d
commit 452e84703c
1 changed files with 1 additions and 1 deletions
--- 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,