detections/engines/inference_engine.pyx

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")

    cdef int get_batch_size(self):
        return self.batch_size

    cdef run(self, input_data):
        raise NotImplementedError("Subclass must implement run")