mirror of
https://github.com/azaion/annotations.git
synced 2026-04-22 12:16:30 +00:00
fix converting model initialization
This commit is contained in:
@@ -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)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user