mirror of
https://github.com/azaion/detections.git
synced 2026-04-22 08:56:32 +00:00
fix some cython code
This commit is contained in:
@@ -2,6 +2,18 @@ cdef class InferenceEngine:
|
||||
def __init__(self, model_bytes: bytes, batch_size: int = 1, **kwargs):
|
||||
self.batch_size = batch_size
|
||||
|
||||
@property
|
||||
def engine_name(self):
|
||||
return "onnx"
|
||||
|
||||
@staticmethod
|
||||
def get_engine_filename():
|
||||
return None
|
||||
|
||||
@staticmethod
|
||||
def convert_from_onnx(bytes onnx_bytes):
|
||||
return onnx_bytes
|
||||
|
||||
cdef tuple get_input_shape(self):
|
||||
raise NotImplementedError("Subclass must implement get_input_shape")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user