From 3bd4c6f82ed54f563c34807bee39938fd37e9ee1 Mon Sep 17 00:00:00 2001 From: Florian Vahl Date: Fri, 27 Oct 2023 16:35:43 +0200 Subject: [PATCH] Fix warning at startup due to uninitilized memory --- yoeo/detect.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/yoeo/detect.py b/yoeo/detect.py index 7f99ac6..6e95cad 100755 --- 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")