Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -313,7 +313,7 @@ def train():
batch_size=batch_size * 2, batch_size=batch_size * 2,
img_size=img_size_test, img_size=img_size_test,
model=model, 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, iou_thres=0.6,
save_json=final_epoch and is_coco, save_json=final_epoch and is_coco,
single_cls=opt.single_cls, single_cls=opt.single_cls,