glenn-jocher commited on
Commit
452e847
1 Parent(s): cce313d

Update train.py (#5014)

Browse files
Files changed (1) hide show
  1. train.py +1 -1
train.py CHANGED
@@ -412,7 +412,7 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary
412
  batch_size=batch_size // WORLD_SIZE * 2,
413
  imgsz=imgsz,
414
  model=attempt_load(f, device).half(),
415
- iou_thres=0.7 if is_coco else 0.6, # best pycocotools results at 0.7
416
  single_cls=single_cls,
417
  dataloader=val_loader,
418
  save_dir=save_dir,
 
412
  batch_size=batch_size // WORLD_SIZE * 2,
413
  imgsz=imgsz,
414
  model=attempt_load(f, device).half(),
415
+ iou_thres=0.65 if is_coco else 0.60, # best pycocotools results at 0.65
416
  single_cls=single_cls,
417
  dataloader=val_loader,
418
  save_dir=save_dir,