From 9bc3e5b2e23eba1d1c5f507422d56991b39b246c Mon Sep 17 00:00:00 2001 From: Yoni Yang <76271912+yoni13@users.noreply.github.com> Date: Fri, 10 Jan 2025 20:20:21 +0800 Subject: [PATCH] Update rknn.py --- machine-learning/app/sessions/rknn.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/machine-learning/app/sessions/rknn.py b/machine-learning/app/sessions/rknn.py index 356a78c30b..2f0e7cc0fd 100644 --- a/machine-learning/app/sessions/rknn.py +++ b/machine-learning/app/sessions/rknn.py @@ -58,10 +58,7 @@ class RknnSession: # log.info(f"Running inference on RKNN model") - - outputs = self.rknn.inference(inputs=[input_data]) - if not outputs: - outputs = self.rknn.inference(inputs=[input_data], data_format='nchw') + outputs = self.rknn.inference(inputs=[input_data], data_format='nchw') # log.info("inputs:") # log.info(input_data)