From 3df1a7c67584ccf9317044b4b0b7ede42ee40cd3 Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Mon, 30 Sep 2024 13:04:11 -0600 Subject: [PATCH] Fix type check --- frigate/object_detection.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/frigate/object_detection.py b/frigate/object_detection.py index a4daeea2d..eaa3b4e04 100644 --- a/frigate/object_detection.py +++ b/frigate/object_detection.py @@ -13,7 +13,7 @@ from setproctitle import setproctitle import frigate.util as util from frigate.detectors import create_detector from frigate.detectors.detector_config import BaseDetectorConfig, InputTensorEnum -from frigate.detectors.plugins.rocm import ROCmDetector +from frigate.detectors.plugins.rocm import DETECTOR_KEY as ROCM_DETECTOR_KEY from frigate.util.builtin import EventsPerSecond, load_labels from frigate.util.image import SharedMemoryFrameManager from frigate.util.services import listen @@ -48,7 +48,7 @@ class LocalObjectDetector(ObjectDetector): self.labels = load_labels(labels) if detector_config: - if detector_config is ROCmDetector: + if detector_config.type == ROCM_DETECTOR_KEY: # ROCm requires NHWC as input self.input_transform = None else: