administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-08 12:15:25 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Add tests

Browse Source
This commit is contained in:
Nicolas Mowen 2024-02-05 15:19:37 -07:00
parent ed572b7954
commit 78ed912a20
2 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
frigate/const.py
View File

@ -27,7 +27,7 @@ LABEL_CONSOLIDATION_MAP = {
} }
LABEL_CONSOLIDATION_DEFAULT = 0.9 LABEL_CONSOLIDATION_DEFAULT = 0.9
LABEL_NMS_MAP = { LABEL_NMS_MAP = {
"car": 0.7, "car": 0.6,
} }
LABEL_NMS_DEFAULT = 0.4 LABEL_NMS_DEFAULT = 0.4

6
frigate/test/test_video.py
View File

@ -287,6 +287,12 @@ class TestObjectBoundingBoxes(unittest.TestCase):
consolidated_detections = reduce_detections(frame_shape, detections) consolidated_detections = reduce_detections(frame_shape, detections)
assert len(consolidated_detections) == len(detections) assert len(consolidated_detections) == len(detections)
def test_vert_stacked_cars_not_reduced(self):
detections = [("car", 0.8, (954, 312, 1247, 475), 498512, 1.48, (800, 200, 1400, 600)), ("car", 0.85, (970, 380, 1273, 610), 698752, 1.56, (800, 200, 1400, 700))]
frame_shape = (720, 1280)
consolidated_detections = reduce_detections(frame_shape, detections)
assert len(consolidated_detections) == len(detections)
class TestRegionGrid(unittest.TestCase): class TestRegionGrid(unittest.TestCase):
def setUp(self) -> None: def setUp(self) -> None: