mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-10-11 17:22:49 +03:00
Josh HawkinsandGitHub
5c32af3c7f
CI / AMD64 Build (push) Waiting to run
CI / AMD64 Smoke Test (push) Blocked by required conditions
CI / ARM Build (push) Waiting to run
CI / Jetson Jetpack 6 (push) Waiting to run
CI / AMD64 Extra Build (push) Blocked by required conditions
CI / ARM Extra Build (push) Blocked by required conditions
CI / Synaptics Build (push) Blocked by required conditions
CI / Assemble and push default build (push) Blocked by required conditions
* fix autotracking tracker selection and crop before histogram conversion - Rebuild the norfair trackers when an onvif save changes autotracking enabled_in_config or the PTZ-tracked labels. The trackers were only built at startup, so disabling autotracking or removing person from its track list in the UI made get_tracker pick a tracker that didn't exist, and the KeyError killed the camera process until restart. Saves that leave the selection alone, and runtime MQTT toggles, don't touch tracking. - get_tracker now returns the default tracker that match_and_update actually feeds. On autotracking cameras it returned the static default for labels without their own tracker, so register missed the norfair object and lost the pre-initialization score history, and deregister pruned the wrong tracker. - get_histogram crops the box out of each I420 plane before converting to BGR instead of converting the whole frame. It runs per detection per frame on autotracking cameras and now costs about 0.2 ms at any resolution, down from 1 ms at 1080p and 5 ms at 4K. * don't rebuild trackers for track list edits while autotracking is disabled
42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
import unittest
|
|
|
|
import cv2
|
|
import numpy as np
|
|
|
|
from frigate.util.image import get_histogram
|
|
|
|
|
|
def reference_histogram(image, x_min, y_min, x_max, y_max):
|
|
bgr = cv2.cvtColor(image, cv2.COLOR_YUV2BGR_I420)[y_min:y_max, x_min:x_max]
|
|
hist = cv2.calcHist([bgr], [0, 1, 2], None, [8, 8, 8], [0, 256, 0, 256, 0, 256])
|
|
return cv2.normalize(hist, hist).flatten()
|
|
|
|
|
|
class TestGetHistogram(unittest.TestCase):
|
|
def setUp(self):
|
|
rng = np.random.default_rng(0)
|
|
self.frame = rng.integers(0, 255, (720 * 3 // 2, 1280), np.uint8)
|
|
|
|
def test_matches_full_frame_conversion_on_even_box(self):
|
|
box = (100, 200, 400, 600)
|
|
np.testing.assert_array_equal(
|
|
get_histogram(self.frame, *box), reference_histogram(self.frame, *box)
|
|
)
|
|
|
|
def test_odd_box_widens_to_even_edges(self):
|
|
np.testing.assert_array_equal(
|
|
get_histogram(self.frame, 101, 201, 399, 599),
|
|
reference_histogram(self.frame, 100, 200, 400, 600),
|
|
)
|
|
|
|
def test_box_is_clamped_to_frame(self):
|
|
np.testing.assert_array_equal(
|
|
get_histogram(self.frame, -10, -10, 5000, 5000),
|
|
reference_histogram(self.frame, 0, 0, 1280, 720),
|
|
)
|
|
|
|
def test_empty_box_returns_zeros(self):
|
|
hist = get_histogram(self.frame, 50, 50, 50, 50)
|
|
self.assertEqual(hist.shape, (512,))
|
|
self.assertEqual(hist.sum(), 0)
|