Ë
    Fêñi+  ã                  óV   — d dl mZ d dlmZ d dlmZmZmZ 	 	 	 d	 	 	 	 	 	 	 	 	 	 	 dd„Zy)é    )Úannotations)ÚPath)ÚIS_COLABÚLOGGERÚYAMLNc           	     ó  — ddl m} t        j                  d|› d�«        |d«        |d«       t        rddl}d„ |_        dd	lm} t        |«      }|j                  d
d
¬«        |d¬«      }|j                  g d¢gg d¢g|¬«       |j                  | ¬«       |j                  d¬«       |j                  t        |t        | «      j                   › d|› d�z  «      «       |rt#        j$                  |dz  |«       t        |«      S )aç  Export an ONNX model to RKNN format for Rockchip NPUs.

    Args:
        onnx_file (str): Path to the source ONNX file (already exported, opset <=19).
        output_dir (Path | str): Directory to save the exported RKNN model.
        name (str): Target platform name (e.g. ``"rk3588"``).
        metadata (dict | None): Metadata saved as ``metadata.yaml``.
        prefix (str): Prefix for log messages.

    Returns:
        (str): Path to the exported ``_rknn_model`` directory.
    r   )Úcheck_requirementsú
z& starting export with rknn-toolkit2...zrknn-toolkit2>=2.3.2zonnx<1.19.0Nc                  ó   — y )N© r   ó    ú_/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/ultralytics/utils/export/rknn.pyú<lambda>zonnx2rknn.<locals>.<lambda>'   s   � r   )ÚRKNNT)ÚparentsÚexist_okF)Úverbose)r   r   r   )éÿ   r   r   )Úmean_valuesÚ
std_valuesÚtarget_platform)Úmodel)Údo_quantizationú-z.rknnzmetadata.yaml)Úultralytics.utils.checksr	   r   Úinfor   ÚbuiltinsÚexitÚrknn.apir   r   ÚmkdirÚconfigÚ	load_onnxÚbuildÚexport_rknnÚstrÚstemr   Úsave)	Ú	onnx_fileÚ
output_dirÚnameÚmetadataÚprefixr	   r   r   Úrknns	            r   Ú	onnx2rknnr.   
   sé   € õ& <ä
‡K�K�"�V�HÐBÐCÔDÙÐ-Ô.Ù�}Ô%åãá$ˆŒåä�jÓ!€JØ×Ñ˜T¨DÐÔ1á˜Ô€DØ‡K�KšY˜K²_Ð4EÐW[€KÔ\Ø‡N�N˜€NÔ#Ø‡J�J˜u€JÔ%Ø×Ñ”S˜¬¨i«×)=Ñ)=Ð(>¸aÀ¸vÀUÐ&KÑKÓLÔMÙÜ�	‰	�*˜Ñ.°Ô9Üˆz‹?Ðr   )Úrk3588NÚ )r(   r%   r)   z
Path | strr*   r%   r+   zdict | Noner,   r%   Úreturnr%   )	Ú
__future__r   Úpathlibr   Úultralytics.utilsr   r   r   r.   r   r   r   ú<module>r5      s[   ðõ #å ç 4Ñ 4ð Ø Øð+Øð+àð+ð ð+ð ð	+ð
 ð+ð 	ô+r   