Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-05-21 01:16:37 +02:00
Code Issues Projects Releases Wiki Activity

fix: use requests.Session() for DeepStack API (#15505)

Browse Source
This commit is contained in:
FL42 2024-12-14 15:54:13 +01:00 committed by GitHub
parent 0763f56047
commit 1b7fe9523d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -32,6 +32,7 @@ class DeepStack(DetectionApi):
self.api_timeout = detector_config.api_timeout self.api_timeout = detector_config.api_timeout
self.api_key = detector_config.api_key self.api_key = detector_config.api_key
self.labels = detector_config.model.merged_labelmap self.labels = detector_config.model.merged_labelmap
self.session = requests.Session()
def get_label_index(self, label_value): def get_label_index(self, label_value):
if label_value.lower() == "truck": if label_value.lower() == "truck":
@ -51,7 +52,7 @@ class DeepStack(DetectionApi):
data = {"api_key": self.api_key} data = {"api_key": self.api_key}
try: try:
response = requests.post( response = self.session.post(
self.api_url, self.api_url,
data=data, data=data,
files={"image": image_bytes}, files={"image": image_bytes},