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

default test --conf to 0.001

This commit is contained in:
Glenn Jocher 2020-03-31 14:36:25 -07:00
parent 98271eb6ed
commit 992e0d7cb4
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -315,8 +315,6 @@ def train():
batch_size=batch_size, batch_size=batch_size,
img_size=img_size_test, img_size=img_size_test,
model=ema.ema, model=ema.ema,
conf_thres=0.001 if final_epoch else 0.01, # 0.001 for best mAP, 0.01 for speed
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,
dataloader=testloader) dataloader=testloader)