Merge pull request #114 from bit-bots/Flova-patch-1

Fix warning at startup due to uninitilized memory
bit-bots · Oct 28, 2023 · 6003b88 · 6003b88
2 parents 5f18491 + 3bd4c6f
commit 6003b88
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/yoeo/detect.py b/yoeo/detect.py
@@ -96,8 +96,8 @@ def detect_image(model, image, img_size=416, conf_thres=0.5, nms_thres=0.5, robo
         DEFAULT_TRANSFORMS,
         Resize(img_size)])((
             image,
-            np.empty((1, 5)),
-            np.empty((img_size, img_size), dtype=np.uint8)))[0].unsqueeze(0)
+            np.zeros((1, 5)),
+            np.zeros((img_size, img_size), dtype=np.uint8)))[0].unsqueeze(0)
 
     if torch.cuda.is_available():
         input_img = input_img.to("cuda")