jtDddlmZddlZddlmZddlmZ dddZdS)) annotationsN)Path)LOGGERF onnx_filestr output_file Path | strhalfboolint8metadata dict | Noneprefixreturnc ddlm}ddlm}|s Jdt |s Jd||dddl}ddlm} tj d |d | d d d dd|dt|dtj|pig } |r| d|r| d| | t |jdz } | r| t|S)aConvert an ONNX model to MNN format. Args: onnx_file (str): Path to the source ONNX file. output_file (Path | str): Path to save the exported MNN model. half (bool): Whether to enable FP16 conversion. int8 (bool): Whether to enable INT8 weight quantization. metadata (dict | None): Optional metadata embedded via ``--bizCode``. prefix (str): Prefix for log messages. Returns: (str): Path to the exported ``.mnn`` file. r)check_requirements) TORCH_1_10z>MNN export requires torch>=1.10.0 to avoid segmentation faultszfailed to export ONNX file: z MNN>=2.9.6N) mnnconvert z starting export with MNN z...rz-fONNXz --modelFilez --MNNModelz --bizCode)z--weightQuantBits8z--fp16z.__convert_external_data.bin)ultralytics.utils.checksrultralytics.utils.torch_utilsrrexistsMNN MNN.toolsrrinfoversionrjsondumpsextendappendconvertparentunlink) rr r r rrrrrrmnn_argsconvert_scratchs _/var/www/html/Carbon-Document/venv/lib/python3.11/site-packages/ultralytics/utils/export/mnn.pyonnx2mnnr* s*<;;;;;888888 WWWWW:  ?? ! ! # #OO%OI%O%OOO #|$$$JJJ$$$$$$ KIVIIs{{}}IIIJJJ   K 8>r"" H 42333 "!!!x   ;''.1OOO!   {  )FFNr)rrr r r r r r rrrrrr) __future__rr pathlibrultralytics.utilsrr*r+r)r0sw#""""" $$$$$$   4444444r+