From 37acbdc0b6ef8c3343560834b914c83bbb0abbd1 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Wed, 15 Jul 2020 20:03:41 -0700
Subject: [PATCH] update test.py --save-txt

---
 test.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/test.py b/test.py
index b68db513df3c..ed7e29caf66a 100644
--- a/test.py
+++ b/test.py
@@ -27,7 +27,12 @@ def test(data,
 
     else:  # called directly
         device = torch_utils.select_device(opt.device, batch_size=batch_size)
-        merge = opt.merge  # use Merge NMS
+        merge, save_txt = opt.merge, opt.save_txt  # use Merge NMS, save *.txt labels
+        if save_txt:
+            out = Path('inference/output')
+            if os.path.exists(out):
+                shutil.rmtree(out)  # delete output folder
+            os.makedirs(out)  # make new output folder
 
         # Remove previous
         for f in glob.glob(str(Path(save_dir) / 'test_batch*.jpg')):