From 992e0d7cb48e99b677a337f942c4376bcd544696 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 31 Mar 2020 14:36:25 -0700 Subject: [PATCH] default test --conf to 0.001 --- train.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/train.py b/train.py index f37eb77b..9df15801 100644 --- 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)