delete some useless params and modify detect.py

zhiqwang · Dec 30, 2021 · 3fe0b57 · 3fe0b57
1 parent aeeb839
commit 3fe0b57
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/detect.py b/tools/detect.py
@@ -102,7 +102,7 @@ def run(
     engine = PredictorTRT(weights, device=device)
     stride, names = engine.stride, engine.names
     img_size = check_img_size(img_size, stride=stride)  # check image size
-    img_size = img_size * 2 if len(img_size) == 1 else img_size
+    img_size = img_size * 2 if len(img_size) == 1 else 1  # auto expand
     # Dataloader
     dataset = LoadImages(source, img_size=img_size, stride=stride, auto=False)