mirror of
https://github.com/azaion/detections.git
synced 2026-04-22 13:06:33 +00:00
Refactor inference and AI configuration handling
- Updated the `Inference` class to replace the `get_onnx_engine_bytes` method with `download_model`, allowing for dynamic model loading based on a specified filename. - Modified the `convert_and_upload_model` method to accept `source_bytes` instead of `onnx_engine_bytes`, enhancing flexibility in model conversion. - Introduced a new property `engine_name` to the `Inference` class for better access to engine details. - Adjusted the `AIRecognitionConfig` structure to include a new method pointer `from_dict`, improving configuration handling. - Updated various test cases to reflect changes in model paths and timeout settings, ensuring consistency and reliability in testing.
This commit is contained in:
@@ -4,9 +4,8 @@ from engines.inference_engine cimport InferenceEngine
|
||||
cdef class CoreMLEngine(InferenceEngine):
|
||||
|
||||
cdef object model
|
||||
cdef str input_name
|
||||
cdef tuple input_shape
|
||||
cdef list _output_names
|
||||
cdef int img_width
|
||||
cdef int img_height
|
||||
|
||||
cdef tuple get_input_shape(self)
|
||||
cdef int get_batch_size(self)
|
||||
|
||||
Reference in New Issue
Block a user