Files
annotations/Azaion.AI/ai_config.pyx
T
Alex Bezdieniezhnykh 62623b7123 add ramdisk, load AI model to ramdisk and start recognition from it
rewrite zmq to DEALER and ROUTER
add GET_USER command to get CurrentUser from Python
all auth is on the python side
inference run and validate annotations on python
2025-01-29 17:45:26 +02:00

32 lines
1.6 KiB
Cython

from msgpack import unpackb
cdef class AIRecognitionConfig:
def __init__(self,
frame_recognition_seconds,
tracking_distance_confidence,
tracking_probability_increase,
tracking_intersection_threshold,
frame_period_recognition,
file_data
):
self.frame_recognition_seconds = frame_recognition_seconds
self.tracking_distance_confidence = tracking_distance_confidence
self.tracking_probability_increase = tracking_probability_increase
self.tracking_intersection_threshold = tracking_intersection_threshold
self.frame_period_recognition = frame_period_recognition
self.file_data = file_data
def __str__(self):
return (f'frame_seconds : {self.frame_recognition_seconds}, distance_confidence : {self.tracking_distance_confidence}, '
f'probability_increase : {self.tracking_probability_increase}, intersection_threshold : {self.tracking_intersection_threshold}, frame_period_recognition : {self.frame_period_recognition}')
@staticmethod
cdef from_msgpack(bytes data):
unpacked = unpackb(data, strict_map_key=False)
return AIRecognitionConfig(
unpacked.get("FrameRecognitionSeconds", 0.0),
unpacked.get("TrackingDistanceConfidence", 0.0),
unpacked.get("TrackingProbabilityIncrease", 0.0),
unpacked.get("TrackingIntersectionThreshold", 0.0),
unpacked.get("FramePeriodRecognition", 0),
unpacked.get("Data", b''))