From b0b19b3b94869689dc7c0c0bd832594ea9fa2b92 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Wed, 21 Nov 2018 18:42:06 +0100
Subject: [PATCH] updates

---
 detect.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/detect.py b/detect.py
index 3c412915..90edd2c0 100755
--- a/detect.py
+++ b/detect.py
@@ -25,7 +25,7 @@ parser.add_argument('-img_size', type=int, default=32 * 13, help='size of each i
 opt = parser.parse_args()
 print(opt)
 
-def detect(opt):
+def main(opt):
     os.system('rm -rf ' + opt.output_folder)
     os.makedirs(opt.output_folder, exist_ok=True)
 
@@ -142,4 +142,4 @@ def detect(opt):
 
 if __name__ == '__main__':
     torch.cuda.empty_cache()
-    detect(opt)
+    main(opt)