fix inference

fix small issues
This commit is contained in:
Alex Bezdieniezhnykh
2025-02-14 09:00:04 +02:00
parent cfd5483a18
commit 961d2499de
15 changed files with 42 additions and 29 deletions
+10 -4
View File
@@ -10,7 +10,8 @@ cdef class AIRecognitionConfig:
tracking_probability_increase,
tracking_intersection_threshold,
file_data
file_data,
paths
):
self.frame_period_recognition = frame_period_recognition
self.frame_recognition_seconds = frame_recognition_seconds
@@ -21,10 +22,14 @@ cdef class AIRecognitionConfig:
self.tracking_intersection_threshold = tracking_intersection_threshold
self.file_data = file_data
self.paths = paths
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}')
f'probability_increase : {self.tracking_probability_increase}, '
f'intersection_threshold : {self.tracking_intersection_threshold}, '
f'frame_period_recognition : {self.frame_period_recognition}, '
f'paths: {self.paths}')
@staticmethod
cdef from_msgpack(bytes data):
@@ -38,5 +43,6 @@ cdef class AIRecognitionConfig:
unpacked.get("TrackingProbabilityIncrease", 0.0),
unpacked.get("TrackingIntersectionThreshold", 0.0),
unpacked.get("Data", b''))
unpacked.get("Data", b''),
unpacked.get("Paths", []),
)