|h% 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@jCdd. e j@jCe xsed/ d0Z2ycc}}}}}}wcc}}}}}}}wcc}}}}}}wcc}}}}}wcc}}}}}}w)1N)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 Q/var/www/html/test/engine/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 ttt|jdd||||t||}trLt |}|j |j|jdtj}t|tg|z|rdndtj|dy ) zQTest YOLO model export to OpenVINO under various configuration matrix conditions.r)r)rrr%int8halfbatchdatanms-@rT) ignore_errorsN)r r rr rrrename with_stemstemuuiduuid4rshutilrmtreetaskr%r0r1r2r4rs r test_export_openvino_matrixrA-s  4 ! ( (   t_  ) DDz{{4>>TYYKq*GHIDJx%Wr"= 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)r@r%r0r1r2rDr4rs r test_export_onnx_matrixrGNsc  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%r0r1r2r4r6rNrEr?s r test_export_torchscript_matrixrJbsa  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%r0r1r2rN)r r rrr=r>)r@r%r0r1r2rs r test_export_coreml_matrixrMtsZ  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.tfliterrIrNrEr?s r test_export_tflite_matrixrPs]$  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.rLTr)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_coremlrYs[[]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.rOrr*rNrmodelrs r test_export_tfliter]s/ KE <r}sw 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 !! !! D!noI&/ST!Up!e@(*8s%!V+,V(4V3 V=%W