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

remove cpdef, add constants h

Browse Source
This commit is contained in:
Oleksandr Bezdieniezhnykh
2025-08-12 14:53:14 +03:00
parent ad782bcbaa
commit 9e4dc5404c
6 changed files with 68 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Azaion.Inference/onnx_engine.pyx
+3 -3
View File
@@ -15,12 +15,12 @@ cdef class OnnxEngine(InferenceEngine):
model_meta = self.session.get_modelmeta()
constants_inf.log(f"Metadata: {model_meta.custom_metadata_map}")
cpdef tuple get_input_shape(self):
cdef tuple get_input_shape(self):
shape = self.input_shape
return shape[2], shape[3]
cpdef int get_batch_size(self):
cdef int get_batch_size(self):
return self.batch_size
cpdef run(self, input_data):
cdef run(self, input_data):
return self.session.run(None, {self.input_name: input_data})