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)