jGddlZddlZddlZddlZddlZddlmZmZddlm Z ddl m Z ddl Z ddl Z ddlmZmZddlmZddlmZmZmZddlmZmZmZmZmZmZmZmZdd l m!Z!dd l"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*d Z+e j,-d d dgdZ.e j,-d d dgdZ/dZ0e j,1e' de j,-d d dgdZ2e j,j3e j,1e' de j,-dde edd gdd gdd gddgdd gdgDdZ4e j,j3e j,-dde edd gd gd gddgdd gdd gdd gDdZ5e j,j3e j,-dde ed dgd gd dgddgdd gdd gDdZ6e j,j3e j,1e de j,1e$ d e j,1ej7d!e j,1eoeoed"kd#e j,-d$d%e edd gdd gdd gdd gdgdd gDd&Z8e j,j3e j,1ej9 pe% d'e j,1e ped(e j,-dd)e ed gdd gdd gdgdd gdd gDd*Z:e j,1e$ d e j,1ed+e j,1eoed,e j,1ej7d!d-Z;e j,1ej9 d.e j,1e% d/e j,1e d0d1Ze j,j3e j,1e# d7d8Z?e j,j3e j,1e# d7e j,-d9d:e edd gdd gddgdd gDd;Z@e j,j3e j,1e& d<d=ZAe j,j3e j,1e& d<e j,-d>eBe edd gdgd?ZCe j,1e) d@e j,1ejD dAe j,1e dBe j,1edCdDZEe j,1ej9 pe) dEe j,1edFdGZFe j,j3e j,1ej9 pe) dEe j,1edFe j,-dHedIZGe j,j3e j,1e( pe*dJe j,1e peoedKe j,1edLdMZHe j,j3e j,1e pedNe j,1ejI dOdPZJdS)QN)redirect_stderrredirect_stdout)product)Path)MODELSOURCE)YOLO) TASK2DATA TASK2MODELTASKS)ARM64 IS_DOCKERIS_RASPBERRYPILINUXMACOS MACOS_VERSIONWINDOWSchecks) torch2onnx) TORCH_1_10 TORCH_1_11 TORCH_1_13 TORCH_2_0 TORCH_2_1 TORCH_2_8 TORCH_2_9 TORCH_2_12cPtr|dkrtjddSdSdS)zRSkip semantic segmentation export tests on Raspberry Pi due to memory constraints.semanticzYSemantic segmentation export tests are skipped on Raspberry Pi due to memory constraints.N)rpytestskip)tasks U/var/www/html/Carbon-Document/venv/lib/python3.11/site-packages/tests/test_exports.pyskip_rpi_semanticr$ s>q$*,, opppppqq,,end2endFTcttddd|}t|tddS)zOTest YOLO model export to TorchScript format for compatibility and correctness. torchscriptF )formatoptimizeimgszr&r,Nr rexportrr&files r#test_export_torchscriptr2&sH ;;  ]U"V]  ^ ^DDJJvR      r%cttddd|}t|tddS)z8Test YOLO model export to ONNX format with dynamic axes.onnxTr))r*dynamicr,r&r-Nr.r0s r#test_export_onnxr6-sG ;;  VTW  U UDDJJvR      r%c ddgtj  fd}|tjd|dt ffd fdt dD}|D]}||D]}|r Jdd ks Jd d S) z9Ensure ONNX exports do not overlap across worker threads.rc5dz tdddn #1swxYwYtjd5dzddddS#1swxYwYdS)Ng?)maxtimesleep)argskwargsactive max_active state_locks r# fake_exportzBtest_torch2onnx_serializes_concurrent_exports..fake_export;s  1 1 aKFZ00J 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 4    aKF                  s&**AA!Ar/indexc  ttjtjddddt d|dz dS#t $r }|Yd}~dSd}~wwxYw)Nr9zexport-z.onnx)rtorchnnIdentityzerosstr Exceptionappend)rCerrorerrorstmp_paths r# export_modelzCtest_torch2onnx_serializes_concurrent_exports..export_modelFs ! ux((**EK1a,C,CSTj^cTjTjTjIjEkEk l l l l l ! ! ! MM%  !sAA B%BBc>g|]}tj|fS))targetr=) threadingThread).0irQs r# zAtest_torch2onnx_serializes_concurrent_exports..Ls+RRRAy|1$???RRRr%zConcurrent export errors: r9z&Expected max 1 concurrent export, got N) rTLocksetattrrGr4intrangestartjoin) monkeypatchrPrBthreadsthreadr?rOrQr@rAs ` @@@@@r#-test_torch2onnx_serializes_concurrent_exportsrc4s' FJ F!!J Hk:::!C!!!!!!! SRRRqRRRG  <<=2.1)reasoncJttdd|}trRt |}|||jdtj }t|tddS)zFTest YOLO export to OpenVINO format for model inference compatibility.openvinor))r*r,r&-r-N) r rr/rrrename with_stemstemuuiduuid4rr0s r#test_export_openvinormVs ;;  Zr7  K KDJDzz{{4>>TY*G*G*G*GHHIIDJJvR      r%z.task, dynamic, int8, half, batch, nms, end2endc Pg|]#\}}}}}}}|r|s|dkr||r||||||||f$SclassifyrVr"r5int8halfbatchnmsr&s r#rXrXfsr :D'4uc7    %)J$6$63$6G$6PS$6 wdE38%7$6$6r%r9c t|tt|dd||||t||| }t rRt |}|||j dtj }t|tg|z|rdnd|tj|dd S) zQTest YOLO model export to OpenVINO under various configuration matrix conditions.rfr)) r*r,r5rsrtrudatarvr&rg@)r,ruT ignore_errorsN)r$r r r/r rrrhrirjrkrlrshutilrmtreer"r5rsrtrurvr&r1s r#test_export_openvino_matrixrbsd  4 ! ! ( (   t_  )  DJDzz{{4>>TY*G*G*G*GHHIIDJJx%W'=1.11z#CoreML not supported in Python 3.1315z.CoreML YOLO26 matrix test crashes on macOS 15+z.task, dynamic, int8, half, nms, batch, end2endc hg|]/\}}}}}}}|r|s!|dkr|s|r|s|dkr|"|r|&|||||||f0S)detectrprq)rVr"r5rsrtrvrur&s r#rXrXs    :D'4sE7    !!c!" "###$ $ wdC8$##r%c t|tt|dd||||||}t|tg|zdt j|dS)zHTest YOLO export to CoreML format with various parameter configurations.coremlr)rr-Nr$r r r/rr}r~)r"r5rsrtrvrur&r1s r#test_export_coreml_matrixrs.d  4 ! ! ( (    )  DDJJx%r**** M$r%z3TFLite export requires Python>=3.10 and torch>=1.13zUTest disabled as TF suffers from install conflicts on Windows, macOS and Raspberry Pic tg|]5\}}}}}}}|r|s'|dkr|tr||rt(|r|,|||||||f6Sro)r rrrs r#rXrXs    :D'4uc7    ""s"###&## # wdE38 #""r%c t|tt|dd||||||}t|tg|zdt |dS)zLTest YOLO export to TFLite format considering various export configurations.tfliter)rr-Nrrs r#test_export_tflite_matrixrs4d  4 ! ! ( (r7DPU[^ho )  DDJJx%r****JJr%zCoreML not supported on Windowsz%CoreML not supported on aarch64 Linuxcptjtj}}t|5t|5t t dddtrHt t dd}t |tddddn #1swxYwYdddn #1swxYwY| | z}d|vs Jd |d |vs Jd dS) z7Test YOLO export to CoreML format and check for errors.rTr))r*rvr,r*r,r-NErrorzCoreML export produced errors: z%You will not be able to run predict()z!CoreML export has predict() error) ioStringIOrrr rr/rrgetvalue)stdoutstderrr1outputs r#test_export_coremlrs[]]BKMMFF  ))/&"9"9)) U (B???  );;%%XR%@@D DJJvR ( ( ( ( ))))))))))))))))))))))))))))))__  !2!2 2F & "LF"L"L 2& @ @ @Be @ @ @ @ @s6C#A:C  C# C C#C C##C'*C'z#TFLite export requires Python>=3.10z"TFLite export requires torch>=1.13zGTest disabled as TF suffers from install conflicts on Windows and macOSctt}|dd}t|tddS)zRTest YOLO export to TFLite format under specific OS and Python version conditions.rr)rr-Nr.modelr1s r#test_export_tflitersE KKE <=1.10cttdd}t|tddS)zaTest YOLO export to MNN format (WARNING: MNN test must precede NCNN test or CI error on Windows).mnnr)rr-Nr.r1s r#test_export_mnnr,sC ;;  U"  5 5DDJJvR      r%z task, int8, half, batch, end2endc0g|]\}}}}}|r| |||||fSrqrq)rVr"rsrtrur&s r#rXrX8sM ,D$eW tT5'*r%ct|tt|dd||||}t|tg|zdt |dS)zITest YOLO export to MNN format considering various export configurations.rr))r*r,rsrtrur&r-Nr)r"rsrtrur&r1s r#test_export_mnn_matrixr4sd  4 ! ! ( (RdQU]bls ( t tDDJJx%r****JJr%z-NCNN inference causes segfault on PyTorch<2.0cttdd}t|tddS)z Test YOLO export to NCNN format.ncnnr)rr-Nr.rs r#test_export_ncnnrFsC ;;  V2  6 6DDJJvR      r%ztask, half, batchct|tt|dd||}t|tg|zdt j|ddS)zJTest YOLO export to NCNN format considering various export configurations.rr))r*r,rtrur-Tr{Nr)r"rtrur1s r#test_export_ncnn_matrixrNsy d  4 ! ! ( (btSX ( Y YDDJJx%r**** M$d++++++r%z IMX export requires torch>=2.9.0zRequires Python>=3.9z"IMX export only supported on LinuxzSTest disabled as IMX export suffers from OOM (Out of Memory) on Raspberry Pi 5 16GBctd}|dd}t|tddS)zTest YOLO export to IMX format.z yolo11n.ptimxr)rr-N)r r/rrs r#test_export_imxrYsG   E <=3.10 and Torch>=2.9.0zSkipping test on Windowscttdd}t|s Jd|t|dz }|s Jd|t|dz }|s Jd|t j|d d S) z,Test YOLO model export to ExecuTorch format. executorchr)rz/ExecuTorch export failed, directory not found: model.ptez ExecuTorch .pte file not found: metadata.yamlz$ExecuTorch metadata.yaml not found: Tr{Nr rr/rexistsr}r~)r1pte_file metadata_files r#test_export_executorchrgs ;;  \  < =3.12 for CI validation due to torch upgradescttdd}t|s Jd|t j|ddS)z!Test YOLO export to DeepX format.deepxr)rz*DeepX export failed, directory not found: Tr{Nrrs r#test_export_deepxrsl ;;  WB  7 7D ::    SS ST S SSS  M$d++++++r%)Krr}rTr;rk contextlibrr itertoolsrpathlibrr rGtestsrr ultralyticsr ultralytics.cfgr r r ultralytics.utilsr rrrrrrrultralytics.utils.export.enginerultralytics.utils.torch_utilsrrrrrrrrr$mark parametrizer2r6rcskipifrmslowrrrIS_PYTHON_3_13rIS_PYTHON_MINIMUM_3_10rrrrrrrrlistrIS_PYTHON_MINIMUM_3_9rrrrIS_PYTHON_MINIMUM_3_12rrqr%r#rs   77777777 8888888888llllllllllllllllllll666666                    qqq UDM22!!32! UDM22!!32! RRRD M*HIIUDM22!!32JI! M*HII4>Eg D%=4-$Au X\W]? ?   ,,  JI,,>HO D%=5'E7QFT5MDRW=[_afZgI I     $4>Eg E4=5'E4=1a&4-RVX]Q^? ?     I&PQQ N+OPPF)2WXX 5m5  5>n4  >Eg D%=4-$u PQsUY[`Ta? ?        YXQPRQ*"%%7Z@u I b4  >Eg E7T5MD%=1#e}tUZm? ?     ! 0 N+OPPG$EFFEOe,STTF)2WXX f fYXUTGFQP f 55>cdd N+OPPI&opp!!qpQPed!D11I&^__!!`_21!D!lmm22nm2  N+LMM!!NM!  N+LMM&07e}tUZm^_ab]cfjlqer0s0sNM M*YZZ!![Z!  M*YZZ,dd7754-RSQT3U3U.V.VWW,,XW[Z, M*LMM44=STTI&JKKp!!LKUTNM !55FYOwxxG$>?? , ,@?yx ,55FYOwxxG$>??'' , ,('@?yx , M/Z8eff I&%%I [N+stt,,ut gf,I&/abb%%.m,,cb ,,,r%