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

Don't fail if deepstack detector times out (#8979)

Browse Source
This commit is contained in:
Nicolas Mowen 2023-12-15 16:25:21 -07:00 committed by GitHub
parent e390533760
commit c35c7da82a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -49,12 +49,18 @@ class DeepStack(DetectionApi):
image.save(output, format="JPEG") image.save(output, format="JPEG")
image_bytes = output.getvalue() image_bytes = output.getvalue()
data = {"api_key": self.api_key} data = {"api_key": self.api_key}
response = requests.post(
self.api_url, try:
data=data, response = requests.post(
files={"image": image_bytes}, self.api_url,
timeout=self.api_timeout, data=data,
) files={"image": image_bytes},
timeout=self.api_timeout,
)
except requests.exceptions.RequestException:
logger.error("Error calling deepstack API")
return np.zeros((20, 6), np.float32)
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: if response_json.get("predictions") is None: