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

Formatting

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

4
frigate/object_detection.py
View File

@ -52,7 +52,9 @@ class LocalObjectDetector(ObjectDetector):
# ROCm requires NHWC as input
self.input_transform = None
else:
self.input_transform = tensor_transform(detector_config.model.input_tensor)
self.input_transform = tensor_transform(
detector_config.model.input_tensor
)
else:
self.input_transform = None