mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-05-06 21:45:32 +03:00
Compare commits
No commits in common. "ab59fdf570185c2fcbda6c5c88d6aa05cde7a2c5" and "8f99733638bd6a70d6d1aea1f73449dfca6d9dd4" have entirely different histories.
ab59fdf570
...
8f99733638
@ -19,6 +19,8 @@ logger = logging.getLogger(__name__)
|
|||||||
|
|
||||||
DETECTOR_KEY = "rknn"
|
DETECTOR_KEY = "rknn"
|
||||||
|
|
||||||
|
supported_socs = ["rk3562", "rk3566", "rk3568", "rk3576", "rk3588", "rk3588s"]
|
||||||
|
|
||||||
supported_models = {
|
supported_models = {
|
||||||
ModelTypeEnum.yologeneric: "^frigate-fp16-yolov9-[cemst]$",
|
ModelTypeEnum.yologeneric: "^frigate-fp16-yolov9-[cemst]$",
|
||||||
ModelTypeEnum.yolonas: "^deci-fp16-yolonas_[sml]$",
|
ModelTypeEnum.yolonas: "^deci-fp16-yolonas_[sml]$",
|
||||||
|
|||||||
@ -189,16 +189,13 @@ def convert_onnx_to_rknn(
|
|||||||
logger.debug("Could not determine SoC type")
|
logger.debug("Could not determine SoC type")
|
||||||
return False
|
return False
|
||||||
|
|
||||||
# Map rk3588s to rk3588 for RKNN toolkit compatibility
|
|
||||||
target_soc = "rk3588" if soc == "rk3588s" else soc
|
|
||||||
|
|
||||||
# Get model config for the specified type
|
# Get model config for the specified type
|
||||||
if model_type not in MODEL_TYPE_CONFIGS:
|
if model_type not in MODEL_TYPE_CONFIGS:
|
||||||
logger.debug(f"Unsupported model type: {model_type}")
|
logger.debug(f"Unsupported model type: {model_type}")
|
||||||
return False
|
return False
|
||||||
|
|
||||||
config = MODEL_TYPE_CONFIGS[model_type].copy()
|
config = MODEL_TYPE_CONFIGS[model_type].copy()
|
||||||
config["target_platform"] = target_soc
|
config["target_platform"] = soc
|
||||||
|
|
||||||
# RKNN toolkit requires .onnx extension, create temporary copy if needed
|
# RKNN toolkit requires .onnx extension, create temporary copy if needed
|
||||||
temp_onnx_path = None
|
temp_onnx_path = None
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user