Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-08-27 13:47:50 +02:00
Code Issues Projects Releases Wiki Activity

Update detect_objects.py

Browse Source
This commit is contained in:
MikEarpp 2020-10-12 10:42:50 +02:00 committed by GitHub
parent fb1931d43a
commit b2ff50ca2c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
detect_objects.py
View File

@ -441,7 +441,7 @@ def main():
def latest_mask(camera_name):
if camera_name in CONFIG['cameras']:
if 'mask' in CONFIG['cameras'][camera_name] :
if 'mask' in CONFIG['cameras'][camera_name]:
mask = CONFIG['cameras'][camera_name]['mask']
frame = latest_frame(camera_name, True)
@ -467,8 +467,17 @@ def main():
return "Inconsistent sizes, {}x{} and {}x{}.".format(frame.shape[0], frame.shape[1], the_mask.shape[0], the_mask.shape[1]), 404
else:
hardness = 0.5
if 'mask_hardness' in CONFIG['cameras'][camera_name]:
try:
hn = float(CONFIG['cameras'][camera_name]['mask_hardness'])
if isinstance(hn, (int, float)) and not isinstance(hn, bool) and hn >= 0 and hn <= 1:
hardness = hn
except ValueError:
pass
# frame and mask fusion
ret, jpg = cv2.imencode('.jpg', cv2.addWeighted(frame, 0.5, the_mask, 0.5, 0))
ret, jpg = cv2.imencode('.jpg', cv2.addWeighted(frame, 1-hardness, the_mask, hardness, 0))
response = make_response(jpg.tobytes())
response.headers['Content-Type'] = 'image/jpg'
return response