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

updates

This commit is contained in:
Glenn Jocher 2019-12-27 10:52:19 -08:00
parent 2cc805edda
commit 56d7261083
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

@ -330,7 +330,7 @@ def train():
img_size=opt.img_size, img_size=opt.img_size,
model=model, model=model,
conf_thres=0.001 if final_epoch else 0.1, # 0.1 for speed conf_thres=0.001 if final_epoch else 0.1, # 0.1 for speed
iou_thres=0.6 if opt.evolve else 0.5, iou_thres=0.6 if final_epoch and is_coco else 0.5,
save_json=final_epoch and is_coco, save_json=final_epoch and is_coco,
dataloader=testloader) dataloader=testloader)