This commit is contained in:
Glenn Jocher 2020-01-31 09:36:28 -08:00
parent f7772c791d
commit d23f721dcf
1 changed files with 1 additions and 1 deletions

View File

@ -313,7 +313,7 @@ def train():
batch_size=batch_size * 2,
img_size=img_size_test,
model=model,
conf_thres=0.001 if final_epoch and is_coco else 0.1, # 0.1 for speed
conf_thres=1E-3 if opt.evolve or (final_epoch and is_coco) else 0.1, # 0.1 faster
iou_thres=0.6,
save_json=final_epoch and is_coco,
single_cls=opt.single_cls,