chris/frigate
1
0
Fork 0
forked from Github/frigate
Code Pull Requests Packages Activity

add global object mask

Browse Source
This commit is contained in:
Blake Blackshear
2021-02-06 06:30:26 -06:00
parent a99f360a64
commit ce90ae343c
3 changed files with 60 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
frigate/test/test_config.py
View File

@@ -160,6 +160,39 @@ class TestConfig(TestCase):
assert('dog' in frigate_config.cameras['back'].objects.filters)
assert(frigate_config.cameras['back'].objects.filters['dog'].threshold == 0.7)
def test_global_object_mask(self):
config = {
'mqtt': {
'host': 'mqtt'
},
'objects': {
'track': ['person', 'dog']
},
'cameras': {
'back': {
'ffmpeg': {
'inputs': [
{ 'path': 'rtsp://10.0.0.1:554/video', 'roles': ['detect'] }
]
},
'height': 1080,
'width': 1920,
'objects': {
'mask': '0,0,1,1,0,1',
'filters': {
'dog': {
'mask': '1,1,1,1,1,1'
}
}
}
}
}
}
frigate_config = FrigateConfig(config=config)
assert('dog' in frigate_config.cameras['back'].objects.filters)
assert(len(frigate_config.cameras['back'].objects.filters['dog']._raw_mask) == 2)
assert(len(frigate_config.cameras['back'].objects.filters['person']._raw_mask) == 1)
def test_ffmpeg_params_global(self):
config = {
'ffmpeg': {