From 3bd4c6f82ed54f563c34807bee39938fd37e9ee1 Mon Sep 17 00:00:00 2001 From: Florian Vahl Date: Fri, 27 Oct 2023 16:35:43 +0200 Subject: [PATCH 1/4] 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") From 106d8b77d36b8f53f841d12be51c7d733137ea19 Mon Sep 17 00:00:00 2001 From: Florian Vahl Date: Thu, 2 Nov 2023 21:09:52 +0100 Subject: [PATCH 2/4] Bump version --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index ec50902..e39a356 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "YOEO" -version = "1.4.1" +version = "1.4.2" description = "A hybrid CNN for object detection and semantic segmentation" authors = ["Florian Vahl ", "Jan Gutsche "] From 8d036e6a552974b81b243851585f9bdd0e75c3ef Mon Sep 17 00:00:00 2001 From: Florian Vahl <7vahl@informatik.uni-hamburg.de> Date: Sat, 2 Dec 2023 15:06:37 +0100 Subject: [PATCH 3/4] Fix tvm args --- yoeo/scripts/convertONNXModelToTVM.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/yoeo/scripts/convertONNXModelToTVM.py b/yoeo/scripts/convertONNXModelToTVM.py index e39b7de..3dfd70d 100644 --- a/yoeo/scripts/convertONNXModelToTVM.py +++ b/yoeo/scripts/convertONNXModelToTVM.py @@ -26,7 +26,6 @@ def make_parser(): type=int, help="Input image size") parser.add_argument( - "-t", "--trials", default=20000, type=int, @@ -56,7 +55,6 @@ def make_parser(): type=str, help="Path to the tuning records that are created for this optimization") parser.add_argument( - "-t", "--target", default="vulkan -from_device=0", type=str, @@ -89,7 +87,7 @@ def run(): # Build an TVM Compiler model tvmc_model = TVMCModel(mod, params) - + # Tune the model (depending on the hardware and parameters this takes days) if not args.no_tuning: tvmc.tune( From 3d58235b7f37836b8a6a9a0945acf890f0735ecc Mon Sep 17 00:00:00 2001 From: Florian Vahl <7vahl@informatik.uni-hamburg.de> Date: Sat, 2 Dec 2023 15:08:48 +0100 Subject: [PATCH 4/4] Bump version --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index e39a356..b5f7e02 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "YOEO" -version = "1.4.2" +version = "1.4.3" description = "A hybrid CNN for object detection and semantic segmentation" authors = ["Florian Vahl ", "Jan Gutsche "]