mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-02-15 07:35:27 +03:00
Formatting
This commit is contained in:
parent
53aa9faf7a
commit
1b5e92e4d3
@ -52,7 +52,9 @@ class LocalObjectDetector(ObjectDetector):
|
|||||||
# ROCm requires NHWC as input
|
# ROCm requires NHWC as input
|
||||||
self.input_transform = None
|
self.input_transform = None
|
||||||
else:
|
else:
|
||||||
self.input_transform = tensor_transform(detector_config.model.input_tensor)
|
self.input_transform = tensor_transform(
|
||||||
|
detector_config.model.input_tensor
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
self.input_transform = None
|
self.input_transform = None
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user