fix converting model initialization

This commit is contained in:
Oleksandr Bezdieniezhnykh
2025-09-04 10:39:41 +03:00
parent b3665630ed
commit be77a81875
8 changed files with 97 additions and 41 deletions
+4 -1
View File
@@ -4,11 +4,12 @@ from annotation cimport Annotation, Detection
from ai_config cimport AIRecognitionConfig
from loader_client cimport LoaderClient
from inference_engine cimport InferenceEngine
from remote_command_handler_inf cimport RemoteCommandHandler
cdef class Inference:
cdef LoaderClient loader_client
cdef InferenceEngine engine
cdef object on_annotation
cdef RemoteCommandHandler remote_handler
cdef Annotation _previous_annotation
cdef dict[str, list(Detection)] _tile_detections
cdef AIRecognitionConfig ai_config
@@ -20,6 +21,7 @@ cdef class Inference:
cdef int model_height
cdef bytes get_onnx_engine_bytes(self)
cdef convert_and_upload_model(self, bytes onnx_engine_bytes, str engine_filename)
cdef init_ai(self)
cdef bint is_building_engine
cdef bint is_video(self, str filepath)
@@ -28,6 +30,7 @@ cdef class Inference:
cdef _process_video(self, RemoteCommand cmd, AIRecognitionConfig ai_config, str video_name)
cdef _process_images(self, RemoteCommand cmd, AIRecognitionConfig ai_config, list[str] image_paths)
cdef _process_images_inner(self, RemoteCommand cmd, AIRecognitionConfig ai_config, list frame_data)
cdef on_annotation(self, RemoteCommand cmd, Annotation annotation)
cdef split_to_tiles(self, frame, path, tile_size, overlap_percent)
cdef stop(self)