mirror of
https://github.com/azaion/detections.git
synced 2026-06-21 15:41:07 +00:00
Add AIAvailabilityStatus and AIRecognitionConfig classes for AI model management
- Introduced `AIAvailabilityStatus` class to manage the availability status of AI models, including methods for setting status and logging messages. - Added `AIRecognitionConfig` class to encapsulate configuration parameters for AI recognition, with a static method for creating instances from dictionaries. - Implemented enums for AI availability states to enhance clarity and maintainability. - Updated related Cython files to support the new classes and ensure proper type handling. These changes aim to improve the structure and functionality of the AI model management system, facilitating better status tracking and configuration handling.
This commit is contained in:
@@ -0,0 +1,20 @@
|
||||
from engines.inference_engine cimport InferenceEngine
|
||||
|
||||
|
||||
cdef class TensorRTEngine(InferenceEngine):
|
||||
|
||||
cdef public object context
|
||||
|
||||
cdef public object d_input
|
||||
cdef public object d_output
|
||||
cdef str input_name
|
||||
cdef list input_shape
|
||||
|
||||
cdef object h_output
|
||||
cdef str output_name
|
||||
cdef list output_shape
|
||||
|
||||
cdef object stream
|
||||
|
||||
cdef tuple get_input_shape(self)
|
||||
cdef run(self, input_data)
|
||||
Reference in New Issue
Block a user