From d23f721dcf9eb26949f07b15f5534ee2918c7fc8 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 31 Jan 2020 09:36:28 -0800 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 15a6d2ad..01c3107e 100644 --- a/train.py +++ b/train.py @@ -313,7 +313,7 @@ def train(): batch_size=batch_size * 2, img_size=img_size_test, 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, save_json=final_epoch and is_coco, single_cls=opt.single_cls,