FiCddlZddlZddlZddlZddlZddlmZmZddlm 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)e jTjWd d d gdZ,e jTjWd d d gdZ-dZ.e jTj_e' de jTjWd d d gdZ0e jTjbe jTj_e' de jTjWde ed d gd d gd d gddgd d gd gDcgc]"\}}}}}}}|r|s|dk(r|s |r|s |||||||f$c}}}}}}}dZ2e jTjbe jTjWde ed d gd gd gddgd d gd d gd d gDc gc](\}}}}}}}}|r|s|dk(r|s|re%r|r|s ||||||||f*c }}}}}}}}dZ3e jTjbe jTjWde ed d gd gd d gddgd d gd d gDcgc]\}}}}}}}|dk(r|s |r|s |||||||f c}}}}}}}dZ4e jTjbe jTj_e de jTj_e$ de jTj_ejjde jTj_exr exredk\d e jTjWd!e ed d gd d gd d gd d gdgd d gDcgc]-\}}}}}}}|r|s|d"k7r|s|r|s|dk(r|s |r|s |||||||f/c}}}}}}}d#Z6e jTjbe jTj_ejn xse% d$e jTj_e xsed%e jTjWde ed gd d gd d gdgd d gd d gDcgc]*\}}}}}}}|r|s|dk(r|ser|s|re%r |r|s |||||||f,c}}}}}}}d&Z8e jTj_e$ de jTj_ed'e jTj_exred(e jTj_ejjdd)Z9e jTj_ejn d*e jTj_e% d+e jTj_e d,d-Z:e jTj_d d.e jTj_e d/d0Z;e jTj_d d1d2Ze jTjbe jTj_e& d7d8Z?e jTjbe jTj_e& d7e jTjWd9e@e ed d gdgd:ZAe jTj_e) d;e jTj_ej d<e jTj_e d=e jTj_ed>d?ZCe jTj_ejn xse) d@e jTj_edAdBZDe jTjbe jTj_ejn xse) d@e jTj_edAe jTjWdCedDZEe jTjbe jTj_e( dEe jTj_e xsexredFe jTj_edGdHZFycc}}}}}}}wcc }}}}}}}}wcc}}}}}}}wcc}}}}}}}wcc}}}}}}}wcc}}}}}w)IN)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_9end2endFTcvttjddd|}t|tdy)zOTest YOLO model export to TorchScript format for compatibility and correctness. torchscriptF )formatoptimizeimgszrr#Nr rexportrrfiles T/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/tests/test_exports.pytest_export_torchscriptr*s2 ;  ]U"V]  ^DDJvR cvttjddd|}t|tdy)z8Test YOLO model export to ONNX format with dynamic axes.onnxTr )r!dynamicr#rr$Nr%r's r)test_export_onnxr/s1 ;  VTW  UDDJvR r+c dd gtj  fd}|jtjd|dt ffd }t dDcgc]}tj||f}}|D]}|j|D]}|jr Jd d k(s Jd y cc}w) z9Ensure ONNX exports do not overlap across worker threads.rc5dz tdddtjd5dzdddy#1swY/xYw#1swYyxYw)Ng?)maxtimesleep)argskwargsactive max_active state_locks r) fake_exportzBtest_torch2onnx_serializes_concurrent_exports..fake_export,sd  1 aKFZ0J 1 4   aKF    1 1  sAAA Ar&indexc  ttjjtjddddt d|dz y#t $r}j|Yd}~yd}~wwxYw)Nr2zexport-z.onnx)rtorchnnIdentityzerosstr Exceptionappend)r<errorerrorstmp_paths r) export_modelzCtest_torch2onnx_serializes_concurrent_exports..export_model7sf ! uxx((*EKK1a,CSV]^c]ddiTjIjEk l ! MM%  !sAA A7A22A7)targetr6zConcurrent export errors: r2z&Expected max 1 concurrent export, got N) threadingLocksetattrr@r-intrangeThreadstartjoin) monkeypatchrIr;rJithreadsthreadr8rHr9r:s ` @@@@r)-test_torch2onnx_serializes_concurrent_exportsrY%s FJ F!J Hk:!C! JOqRAy|1$?RGR  <3F8<<: ?QDZLQQ?Ss"C zOpenVINO requires torch>=2.1)reasoncttjdd|}trLt |}|j |j |jdtj}t|tdy)zFTest YOLO export to OpenVINO format for model inference compatibility.openvinor )r!r#r-r$N) r rr&rrrename with_stemstemuuiduuid4rr's r)test_export_openvinorcGsg ;  Zr7  KDDz{{4>>TYYKq*GHIDJvR r+z.task, dynamic, int8, half, batch, nms, end2endr2classifyc xtt|jdd||||t||| }trLt |}|j |j|jdtj}t|tg|z|rdnd|tj|dy ) zQTest YOLO model export to OpenVINO under various configuration matrix conditions.r\r ) r!r#r.int8halfbatchdatanmsrr]@)r#riT ignore_errorsN)r r r&r rrr^r_r`rarbrshutilrmtreetaskr.rgrhrirkrr(s r)test_export_openvino_matrixrsSs  4 ! ( (   t_  ) DDz{{4>>TYYKq*GHIDJx%Wr"EJ MM$d+r+z8task, dynamic, int8, half, batch, simplify, nms, end2endc tt|jdd||||||| }t|tg|z|rdndt |j y)zKTest YOLO export to ONNX format with various configurations and parameters.r-r ) r!r#r.rgrhrisimplifyrkrrlr$Nr r r&rrunlink) rrr.rgrhrirurkrr(s r)test_export_onnx_matrixrxusk  4 ! ( (    ) DDJx%Wr"=Jr+c tt|jdd||||||}t|tg|z|rdndt |j y)zITest YOLO model export to TorchScript format under varied configurations.rr r!r#r.rgrhrirkrrlr$Nrvrqs r)test_export_torchscript_matrixr{sd  4 ! ( (BdUZ`cmt ) DDJx%Wr"=Jr+z(CoreML inference only supported on macOSz"CoreML export requires torch>=1.11z#CoreML not supported in Python 3.1315z.CoreML YOLO26 matrix test crashes on macOS 15+z.task, dynamic, int8, half, nms, batch, end2enddetectc tt|jdd||||||}t|tg|zdt j |y)zHTest YOLO export to CoreML format with various parameter configurations.coremlr rzr$Nr r r&rrorp)rrr.rgrhrkrirr(s r)test_export_coreml_matrixrs`.  4 ! ( (    ) DDJx%r* MM$r+z3TFLite export requires Python>=3.10 and torch>=1.13zUTest disabled as TF suffers from install conflicts on Windows, macOS and Raspberry Pic tt|jdd||||||}t|tg|zdt |j y)zLTest YOLO export to TFLite format considering various export configurations.tfliter rzr$Nrvrqs r)test_export_tflite_matrixrs`4  4 ! ( (r7DPU[^ho ) DDJx%r*Jr+zCoreML not supported on Windowsz%CoreML not supported on aarch64 Linuxctjtj}}t|5t|5t t j dddtr7t t j dd}t |tddddddd|j|jz}d|vs Jd |d |vsJd y#1swYLxYw#1swYPxYw) z7Test YOLO export to CoreML format and check for errors.rTr )r!rkr#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)stdoutstderrr(outputs r)test_export_coremlrs[[]BKKMFF  )/&"9) U (B? ;%%XR%@D DJvR ( ))__ !2 2F & L$CF8"LL 2& @eBee @))))s$ C6AC*C6*C3 /C66C?z#TFLite export requires Python>=3.10z"TFLite export requires torch>=1.13zGTest disabled as TF suffers from install conflicts on Windows and macOScvtt}|jdd}t|tdy)zRTest YOLO export to TFLite format under specific OS and Python version conditions.rr rr$Nr%modelr(s r)test_export_tfliters/ KE <=1.10crttjdd}t|tdy)zaTest YOLO export to MNN format (WARNING: MNN test must precede NCNN test or CI error on Windows).mnnr rr$Nr%r(s r)test_export_mnnrs- ;  U"  5DDJvR r+z task, int8, half, batch, end2endctt|jdd||||}t|tg|zdt |j y)zITest YOLO export to MNN format considering various export configurations.rr )r!r#rgrhrirr$Nrv)rrrgrhrirr(s r)test_export_mnn_matrixr sU  4 ! ( (RdQU]bls ( tDDJx%r*Jr+z-NCNN inference causes segfault on PyTorch<2.0crttjdd}t|tdy)z Test YOLO export to NCNN format.ncnnr rr$Nr%rs r)test_export_ncnnr1s- ;  V2  6DDJvR r+ztask, half, batchctt|jdd||}t|tg|zdt j |dy)zJTest YOLO export to NCNN format considering various export configurations.rr )r!r#rhrir$TrmNr)rrrhrir(s r)test_export_ncnn_matrixr9sM  4 ! ( (btSX ( YDDJx%r* MM$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 16GBcntd}|jdd}t|tdy)zTest YOLO export to IMX format.z yolo11n.ptimxr rr$N)r r&rrs r)test_export_imxrCs0  E <=3.10 and Torch>=2.9.0zSkipping test on WindowscXttjdd}t|j s Jd|t|dz }|j s Jd|t|dz }|j s Jd|t j |d y ) 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: TrmNr rr&rexistsrorp)r(pte_file metadata_files r)test_export_executorchrQs ;  \  =2.8.0zNAxelera export is only supported on Linux and is not supported on ARM64 DockerzFTest disabled due to OOM (Out of Memory) issues on Raspberry Pi 5 16GBcttjddd}t|j s Jd|t j |dy) z#Test YOLO export to Axelera format.axelerarlz coco8.yaml)r!r#rjz,Axelera export failed, directory not found: TrmNrrs r)test_export_axelerarssP ;  Yb|  LD :   U"Ntf UU  MM$d+r+)GrrorMr4ra contextlibrr itertoolsrpathlibrpytestr@testsrr ultralyticsr ultralytics.cfgr r r ultralytics.utilsr rrrrrrrultralytics.utils.export.enginerultralytics.utils.torch_utilsrrrrrrrmark parametrizer*r/rYskipifrcslowrsrxr{IS_PYTHON_3_13rIS_PYTHON_MINIMUM_3_10rrrrrrrrlistrIS_PYTHON_MINIMUM_3_9rrrr)rrr.rgrhrirkrrus00000000r)rs   7 88lll6xxxUDM2!3! UDM2!3! RD M*HIUDM2!3J! M*HI4?F D%=4-$Au X\W]?  :D'4uc7$DJ$63GPS wdE38 , J,*>IP D%=5'E7QFT5MDRW=[_afZgI  DD'4uhW$DJ$63CPZ`glo wdE8S'B  "4?F E4=5'E4=1a&4-RVX]Q^?  :D'4uc7#S wdE38  I&PQ N+OPF))2WX 5m5  5>n4?F D%=4-$u PQsUY[`Ta?    :D'4sE7!cS#S wdC8     YQR*  %%%7Z@u I b4?F E7T5MD%=1#e}tUZm?    :D'4uc7d "s#JC wdE38   ! 0 N+OPG$EFEOe,STF))2WX fYUGQ f 555>cd N+OPI&op!qQe!D1I&^_!`2!D!lm2n2  N+LM!N!  N+LM&18e}tUZm^_ab]cfjlqer0s ,D$eW tT5'*N M*YZ![!  M*YZ,d754-RSQT3U.VW,X[, M*LM444=STI&JKp!LUN !555FYOwxG$>? ,@y ,555FYOwxG$>?' ,(@y , M*PQ I&%%I [N+st,u R,G B84  L  `s$.'f4?-f?#g 2g /g!$g,