move python inference to Azaion.Inference folder

This commit is contained in:
Alex Bezdieniezhnykh
2025-02-06 10:48:03 +02:00
parent 739759628a
commit ba3e3b4a55
39 changed files with 313 additions and 173 deletions
+27
View File
@@ -0,0 +1,27 @@
from remote_command cimport RemoteCommand
from annotation cimport Annotation
from ai_config cimport AIRecognitionConfig
cdef class Inference:
cdef object session
cdef object on_annotation
cdef Annotation _previous_annotation
cdef AIRecognitionConfig ai_config
cdef bint stop_signal
cdef str model_input
cdef int model_width
cdef int model_height
cdef bint is_video(self, str filepath)
cdef run_inference(self, RemoteCommand cmd, int batch_size=?)
cdef _process_video(self, RemoteCommand cmd, int batch_size)
cdef _process_image(self, RemoteCommand cmd)
cdef stop(self)
cdef preprocess(self, frame)
cdef postprocess(self, output, int img_width, int img_height)
cdef detect_frame(self, frame, long time)
cdef bint is_valid_annotation(self, Annotation annotation)