h[(l ddlZddlZddlZddlmZmZddlmZddlm Z ddl Z ddl m Z m Z ddlmZddlmZmZmZddlmZmZmZmZmZmZdd lmZmZd Zd Ze j@jCe d dZ"e j@jFe j@jCe d e j@jIdeeddgddgddgddgddgDcgc]\}}}}}}|r|s|dk(r|s||||||fc}}}}}}dZ%e j@jFe j@jIdeeddgdgdgddgddgddgDcgc]'\}}}}}}}|r|s|dk(r|s|dk(r|rer |||||||f)c}}}}}}}dZ&e j@jFe j@jIdeedgdgdgddgddgDcgc]\}}}}}}|dk(r|s||||||fc}}}}}}dZ'e j@jFe j@jCe d e j@jCe d e j@jCejPd e j@jIdeedgddgddgdgDcgc]\}}}}}|r|s|||||fc}}}}}dZ)e j@jFe j@jCejT d e j@jCe xsed e j@jIdeedgddgddgdgddgDcgc] \}}}}}}|r|s|dk(r|s er|s||||||f"c}}}}}}d!Z+e j@jCe d e j@jCed" e j@jCexred# e j@jCejPd d$Z,e j@jCejT d e j@jCe d% d&Z-e j@jCdd' e j@jCe d( d)Z.e j@jCdd* d+Z/e j@jFd,Z0e j@jFd-Z1e j@jFe j@jId.eeddgdgDcgc] \}}}|||f c}}}d/Z2e j@jCdd0 e j@jCe xsed1 d2Z3ycc}}}}}}wcc}}}}}}}wcc}}}}}}wcc}}}}}wcc}}}}}}wcc}}}w)3N)redirect_stderrredirect_stdout)product)Path)MODELSOURCE)YOLO) TASK2DATA TASK2MODELTASKS)ARM64IS_RASPBERRYPILINUXMACOSWINDOWSchecks) TORCH_1_9 TORCH_1_13ctttjddd}t|tdy)zOTest YOLO model export to TorchScript format for compatibility and correctness. torchscriptF )formatoptimizeimgszrNr rexportrfiles P/var/www/html/eduruby.in/venv/lib/python3.12/site-packages/tests/test_exports.pytest_export_torchscriptr!s- ;  ]U"  MDDJvR ctttjddd}t|tdy)z8Test YOLO model export to ONNX format with dynamic axes.onnxTr)rdynamicrrNrrs r test_export_onnxr& s- ;  VT  DDDJvR r"zOpenVINO requires torch>=1.13)reasoncrttjdd}t|tdy)zFTest YOLO export to OpenVINO format for model inference compatibility.openvinorrrrNrrs r test_export_openvinor+&s- ;  Zr  :DDJvR r"z%task, dynamic, int8, half, batch, nmsTFclassifyc vtt|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)rrr%int8halfbatchdatanms-@)rr2T ignore_errorsN)r r rr rrrename with_stemstemuuiduuid4rshutilrmtreetaskr%r0r1r2r4rs r test_export_openvino_matrixrB-s  4 ! ( (   t_  ) DDz{{4>>TYYKq*GHIDJx%Wr"EJ MM$d+r"z/task, dynamic, int8, half, batch, simplify, nmsobbc tt|jdd||||||}t|tg|z|rdndt |j y)zKTest YOLO export to ONNX format with various configurations and parameters.r$r)rrr%r0r1r2simplifyr4r6rNr r rrrunlink)rAr%r0r1r2rEr4rs r test_export_onnx_matrixrHNsc  4 ! ( (Rt$e^flo ) DDJx%Wr"=Jr"c tt|jdd|||||}t|tg|z|rdndt |j y)zITest YOLO model export to TorchScript format under varied configurations.rrrrr%r0r1r2r4r6rNrFr@s r test_export_torchscript_matrixrKbsa  4 ! ( (BdUZ`c ) DDJx%Wr"=Jr"z(CoreML inference only supported on macOSz+CoreML>=7.2 not supported with PyTorch<=1.8z#CoreML not supported in Python 3.13z task, dynamic, int8, half, batchctt|jdd||||}t|tg|zdt j |y)zHTest YOLO export to CoreML format with various parameter configurations.coremlr)rrr%r0r1r2rNr r rrr>r?)rAr%r0r1r2rs r test_export_coreml_matrixrOtsZ  4 ! ( (   ) DDJx%r* MM$r"z#TFLite export requires Python>=3.10zUTest 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.tfliterrJrNrFr@s r test_export_tflite_matrixrRs]$  4 ! ( (r7DPU[^ ) 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.rMTr)rr4rr*rNErrorzCoreML export produced errors: z%You will not be able to run predict()z!CoreML export has predict() error) ioStringIOrrr rrrrgetvalue)stdoutstderrroutputs r test_export_coremlr[s[[]BKKMFF  )/&"9) U (B? ;%%XR%@D DJvR ( ))__ !2 2F & L$CF8"LL 2& @eBee @))))s$ C6AC*C6*C3 /C66C?zGTest 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.rQrr*rNrmodelrs r test_export_tfliter_s/ KE <r< contextlibrr itertoolsrpathlibrpytesttestsrr ultralyticsr ultralytics.cfgr r r ultralytics.utilsr rrrrrultralytics.utils.torch_utilsrrr!r&markskipifr+slow parametrizerBrHrKIS_PYTHON_3_13rOIS_PYTHON_MINIMUM_3_10rRr[r_rbrfrirlrnrq)rAr%r0r1r2r4rEs0000000r rs 7 88@! !  N+JK!L!  N+JK+6= D%=4-$Au 6  1D'4uc$DJ$63 wdE3/ , L,*5@G D%=5'E7QFT5MDRW=@  ;D'4uh$DJ$63DEMVYbl wdE8S9  +6=UUGeWW\V]`acd_ehlnsgt5u 1D'4uc "s wdE3/I&PQ M*WXF))2WX&18wu X\^cWdghfi0j ,D'4u wdE* YYR 555>cd I b+6= E7T5MD%=1#e}6  1D'4uc$DJ$63Ec wdE3/   e  M*WXG$EFEOe,STF))2WX fYUGY f 555>cdI&op!qe!D1I&^_!`2!D!lm2n2 !! !! 5