default test --conf to 0.001

2020-03-31 14:36:25 -07:00 · 2020-03-31 14:36:25 -07:00 · 992e0d7cb4
commit 992e0d7cb4
parent 98271eb6ed
1 changed files with 0 additions and 2 deletions
--- a/train.py
+++ b/train.py
@ -315,8 +315,6 @@ def train():
                                      batch_size=batch_size,
                                      img_size=img_size_test,
                                      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,
                                      single_cls=opt.single_cls,
                                      dataloader=testloader)