mirror of
https://github.com/MarcA711/frigate-rockchip.git
synced 2025-10-16 05:40:48 +08:00
Add support for yolox models to onnx detector (#17773)
This commit is contained in:
@@ -77,6 +77,8 @@ class LocalObjectDetector(ObjectDetector):
|
||||
if self.dtype == InputDTypeEnum.float:
|
||||
tensor_input = tensor_input.astype(np.float32)
|
||||
tensor_input /= 255
|
||||
elif self.dtype == InputDTypeEnum.float_denorm:
|
||||
tensor_input = tensor_input.astype(np.float32)
|
||||
|
||||
return self.detect_api.detect_raw(tensor_input=tensor_input)
|
||||
|
||||
|
Reference in New Issue
Block a user