FiRddlmZddlmZddlZddlmZddlmZGddeZ y) ) annotations)PathN)check_requirements) BaseBackendc eZdZdZddZddZy)AxeleraBackendzAxelera AI inference backend for Axelera Metis AI accelerators. Loads compiled Axelera models (.axm files) and runs inference using the Axelera AI runtime SDK. c ddlm}ddlm}t |}t |j dd}|td||jt|j|_ |jd z }|jr'dd lm}|j!|j|yy#t$rtddYwxYw) 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.utilsr apply_metadata)selfweightr wfound metadata_filer s a/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/ultralytics/nn/backends/axelera.py load_modelzAxeleraBackend.load_models   * ' LQWWW%t, =#&=aS$AB BWWSZ(224  6    ! .    - 8 9 "#  #m  sB88CCc@|j|jS)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)rims r forwardzAxeleraBackend.forward2szz"&&(##N)rz str | PathreturnNone)r$z torch.Tensorr'list)__name__ __module__ __qualname____doc__r!r%r&r r r s :< $r&r ) __future__rpathlibrtorchultralytics.utils.checksrbaserr r.r&r r4s"# 7-$[-$r&