Unverified 提交 452e8470 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Update train.py (#5014)

上级 cce313d6
......@@ -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,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论