From 671747318dfdab58857e8afbf8082406335a9f50 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Tue, 10 Sep 2019 14:25:56 +0200
Subject: [PATCH] updates

---
 detect.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/detect.py b/detect.py
index 8b9e4168..df79f625 100644
--- a/detect.py
+++ b/detect.py
@@ -15,7 +15,6 @@ def detect(save_txt=False, save_img=False, stream_img=False):
 
     # Initialize
     device = torch_utils.select_device(force_cpu=ONNX_EXPORT)
-    torch.backends.cudnn.benchmark = False  # set True to speed up constant image size inference
     if os.path.exists(out):
         shutil.rmtree(out)  # delete output folder
     os.makedirs(out)  # make new output folder
@@ -49,6 +48,7 @@ def detect(save_txt=False, save_img=False, stream_img=False):
     # Set Dataloader
     vid_path, vid_writer = None, None
     if streams:
+        torch.backends.cudnn.benchmark = True  # set True to speed up constant image size inference
         dataset = LoadStreams(source, img_size=img_size, half=half)
     elif webcam:
         stream_img = True