Skip to content

Commit

Permalink
Update DeepStack detector to set width and height from input image si…
Browse files Browse the repository at this point in the history
  • Loading branch information
skrashevich committed May 17, 2023
1 parent 5fb96c7 commit b403882
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions frigate/detectors/plugins/deepstack.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,9 +33,6 @@ def __init__(self, detector_config: DeepstackDetectorConfig):
self.api_key = detector_config.api_key
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):
if label_value.lower() == "truck":
label_value = "car"
Expand All @@ -47,6 +44,7 @@ def get_label_index(self, label_value):
def detect_raw(self, tensor_input):
image_data = np.squeeze(tensor_input).astype(np.uint8)
image = Image.fromarray(image_data)
self.w, self.h = image.size
with io.BytesIO() as output:
image.save(output, format="JPEG")
image_bytes = output.getvalue()
Expand Down

0 comments on commit b403882

Please sign in to comment.