jZddlmZddlmZddlZddlmZddlmZGddeZ dS) ) annotations)PathN)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. weight str | PathreturnNonec ddlm}n!#t$rtddYnwxYwddlm}t |}t |dd}|td||t| |_ |j d z }| r0dd lm}|||dSdS) 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. r)opzaxelera-rt==1.6.0zV--extra-index-url https://software.axelera.ai/artifactory/api/pypi/axelera-pypi/simple)cmdsz*.axmNzNo .axm file found in: z metadata.yaml)YAML)axelera.runtimer ImportErrorrrnextrglobFileNotFoundErrorloadstr optimizedmodelparentexistsultralytics.utilsrapply_metadata)selfr rwfound metadata_filers b/var/www/html/Carbon-Document/venv/lib/python3.11/site-packages/ultralytics/nn/backends/axelera.py load_modelzAxeleraBackend.load_models=   * * * * * * *    #m        '&&&&& LLQWWW%%t,, =#$Aa$A$ABB BWWSZZ((2244  6    ! ! : . . . . . .    - 8 8 9 9 9 9 9 : :s  ''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)rr%s r#forwardzAxeleraBackend.forward2szz"&&((###N)r r r r )r%r&r r')__name__ __module__ __qualname____doc__r$r*r+r#r r sF ::::< $ $ $ $ $ $r+r ) __future__rpathlibrtorchultralytics.utils.checksrbaserr r0r+r#r6s#""""" 777777-$-$-$-$-$[-$-$-$-$-$r+