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: