merge ultralytics:master

detect.py

@@ -47,6 +47,7 @@ def detect(save_img=False):
         import tensorflow as tf
         from tensorflow import keras
 
+        stride = None
         with open('data/coco.yaml') as f:
             names = yaml.load(f, Loader=yaml.FullLoader)['names']  # class names (assume COCO)
 
@@ -95,10 +96,9 @@ def _imports_graph_def():
     if webcam:
         view_img = check_imshow()
         cudnn.benchmark = True  # set True to speed up constant image size inference
-        dataset = LoadStreams(source, img_size=imgsz, stride=stride if backend == 'pytorch' else None, auto=backend == 'pytorch')
+        dataset = LoadStreams(source, img_size=imgsz, stride=stride, auto=backend == 'pytorch')
     else:
-        save_img = True
-        dataset = LoadImages(source, img_size=imgsz, stride=stride if backend == 'pytorch' else None, auto=backend == 'pytorch')
+        dataset = LoadImages(source, img_size=imgsz, stride=stride, auto=backend == 'pytorch')
 
     # Get names and colors
     colors = [[random.randint(0, 255) for _ in range(3)] for _ in names]