administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-15 07:35:27 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Fix type check

Browse Source
This commit is contained in:
Nicolas Mowen 2024-09-30 13:04:11 -06:00
parent 1b5e92e4d3
commit 3df1a7c675
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
frigate/object_detection.py
View File

@ -13,7 +13,7 @@ from setproctitle import setproctitle
import frigate.util as util import frigate.util as util
from frigate.detectors import create_detector from frigate.detectors import create_detector
from frigate.detectors.detector_config import BaseDetectorConfig, InputTensorEnum 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.builtin import EventsPerSecond, load_labels
from frigate.util.image import SharedMemoryFrameManager from frigate.util.image import SharedMemoryFrameManager
from frigate.util.services import listen from frigate.util.services import listen
@ -48,7 +48,7 @@ class LocalObjectDetector(ObjectDetector):
self.labels = load_labels(labels) self.labels = load_labels(labels)
if detector_config: if detector_config:
if detector_config is ROCmDetector: if detector_config.type == ROCM_DETECTOR_KEY:
# ROCm requires NHWC as input # ROCm requires NHWC as input
self.input_transform = None self.input_transform = None
else: else: