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

Update DeepStack detector to set width and height from input image size (#6429)

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

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

@ -33,9 +33,6 @@ class DeepStack(DetectionApi):
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.h = detector_config.model.height
self.w = detector_config.model.width
def get_label_index(self, label_value): def get_label_index(self, label_value):
if label_value.lower() == "truck": if label_value.lower() == "truck":
label_value = "car" label_value = "car"
@ -47,6 +44,7 @@ class DeepStack(DetectionApi):
def detect_raw(self, tensor_input): def detect_raw(self, tensor_input):
image_data = np.squeeze(tensor_input).astype(np.uint8) image_data = np.squeeze(tensor_input).astype(np.uint8)
image = Image.fromarray(image_data) image = Image.fromarray(image_data)
self.w, self.h = image.size
with io.BytesIO() as output: with io.BytesIO() as output:
image.save(output, format="JPEG") image.save(output, format="JPEG")
image_bytes = output.getvalue() image_bytes = output.getvalue()