Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2024-11-21 19:07:46 +01:00
Code Issues Projects Releases Wiki Activity

Fix error in parsing DeepStack response JSON and handle cases where predictions field is missing (#6463)

Browse Source
This commit is contained in:
Sergey Krashevich 2023-05-15 15:37:34 +03:00 committed by GitHub
parent e3b9998879
commit 181b53a55d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
frigate/detectors/plugins/deepstack.py
View File

@ -56,8 +56,11 @@ class DeepStack(DetectionApi):
) )
response_json = response.json() response_json = response.json()
detections = np.zeros((20, 6), np.float32) detections = np.zeros((20, 6), np.float32)
if response_json.get("predictions") is None:
logger.debug(f"Error in parsing response json: {response_json}")
return detections
for i, detection in enumerate(response_json["predictions"]): for i, detection in enumerate(response_json.get("predictions")):
logger.debug(f"Response: {detection}") logger.debug(f"Response: {detection}")
if detection["confidence"] < 0.4: if detection["confidence"] < 0.4:
logger.debug(f"Break due to confidence < 0.4") logger.debug(f"Break due to confidence < 0.4")