azaion/annotations
1
0
Fork 0
mirror of https://github.com/azaion/annotations.git synced 2026-04-22 17:46:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

fixed console Log

Browse Source 
fix same files problem in python different libs
correct command logging in command handler
This commit is contained in:
Alex Bezdieniezhnykh
2025-06-14 21:01:32 +03:00
parent 09cfcdf61a
commit c0f8dd792d
29 changed files with 74 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Azaion.Inference/onnx_engine.pyx
+3 -3
View File
@@ -1,6 +1,6 @@
from inference_engine cimport InferenceEngine
import onnxruntime as onnx
cimport constants
cimport constants_inf
cdef class OnnxEngine(InferenceEngine):
def __init__(self, model_bytes: bytes, batch_size: int = 1, **kwargs):
@@ -11,9 +11,9 @@ cdef class OnnxEngine(InferenceEngine):
self.input_name = self.model_inputs[0].name
self.input_shape = self.model_inputs[0].shape
self.batch_size = self.input_shape[0] if self.input_shape[0] != -1 else batch_size
constants.log(f'AI detection model input: {self.model_inputs} {self.input_shape}')
constants_inf.log(f'AI detection model input: {self.model_inputs} {self.input_shape}')
model_meta = self.session.get_modelmeta()
constants.log(f"Metadata: {model_meta.custom_metadata_map}")
constants_inf.log(f"Metadata: {model_meta.custom_metadata_map}")
cpdef tuple get_input_shape(self):
shape = self.input_shape