/j nddlmZddlZddlmZddlZddlmZddlm Z ddl m Z Gdd e Z dS) ) annotationsN)Path)LOGGER)check_requirements) BaseBackendc"eZdZdZd dZd d Zd S)AxeleraBackendzAxelera AI inference backend for Axelera Metis AI accelerators. Loads compiled Axelera models (.axm files) and runs inference using the Axelera AI runtime SDK. Requires the Axelera runtime environment to be activated before use. weight str | PathreturnNonec@tjdstjd ddlm}n'#t$rtddddlm}YnwxYwt|}t| dd }|td || t||_|jd z }|r0dd lm}|| |d Sd S) zLoad an Axelera model from a directory containing a .axm file. Args: weight (str | Path): Path to the Axelera model directory containing the .axm binary. AXELERA_RUNTIME_DIRzAxelera runtime environment is not activated. Please run: source /opt/axelera/sdk/latest/axelera_activate.sh If this fails, verify driver installation: https://docs.ultralytics.com/integrations/axelera/#axelera-driver-installationr)opzaxelera_runtime2==0.1.2zN--extra-index-url https://software.axelera.ai/artifactory/axelera-runtime-pypi)cmdsz*.axmNzNo .axm file found in: z metadata.yaml)YAML)osenvirongetrwarningaxelera.runtimer ImportErrorrrnextrglobFileNotFoundErrorloadstrmodelparentexistsultralytics.utilsrapply_metadata)selfr rwfound metadata_filers d/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/ultralytics/nn/backends/axelera.py load_modelzAxeleraBackend.load_modelsb z~~344  Na    + * * * * * * * + + + )e     + * * * * * * *  + LLQWWW%%t,, =#$Aa$A$ABB BWWSZZ((  6    ! ! : . . . . . .    - 8 8 9 9 9 9 9 : :s<!A A im torch.TensorlistcP||S)zRun inference on the Axelera hardware accelerator. Args: im (torch.Tensor): Input image tensor in BCHW format, normalized to [0, 1]. Returns: (list): Model predictions as a list of output arrays. )rcpu)r$r*s r(forwardzAxeleraBackend.forward<szz"&&((###N)r r r r)r*r+r r,)__name__ __module__ __qualname____doc__r)r/r0r(r r sG #:#:#:#:J $ $ $ $ $ $r0r ) __future__rrpathlibrtorchr"rultralytics.utils.checksrbaserr r5r0r(r;s#"""""  $$$$$$7777775$5$5$5$5$[5$5$5$5$5$r0