From 3c6b168a0a3b799962bb06586e4c84e507af5c85 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sun, 14 Apr 2019 23:22:35 +0200
Subject: [PATCH] updates

---
 test.py  | 2 +-
 train.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/test.py b/test.py
index 5536eab0..222f829a 100644
--- a/test.py
+++ b/test.py
@@ -15,7 +15,7 @@ def test(
         batch_size=16,
         img_size=416,
         iou_thres=0.5,
-        conf_thres=0.1,
+        conf_thres=0.001,
         nms_thres=0.5,
         save_json=False,
         model=None
diff --git a/train.py b/train.py
index ba134bc4..94e5e1e7 100644
--- a/train.py
+++ b/train.py
@@ -180,7 +180,7 @@ def train(
 
         # Calculate mAP
         with torch.no_grad():
-            results = test.test(cfg, data_cfg, batch_size=batch_size, img_size=img_size, model=model)
+            results = test.test(cfg, data_cfg, batch_size=batch_size, img_size=img_size, model=model, conf_thres=0.1)
 
         # Write epoch results
         with open('results.txt', 'a') as file: