.h[+Z ddlZddlZddlZddlmZmZddlmZddlm Z ddl Z ddl m Z m Z ddlmZddlmZmZmZddlmZmZmZmZmZmZdd lmZmZmZd Zd Z e jBjEe d dZ#e jBjHe jBjEe d e jBjKdeeddgddgddgddgddgDcgc]\}}}}}}|r|s|dk(r|s||||||fc}}}}}}dZ&e jBjHe jBjKdeeddgdgdgddgddgddgDcgc]"\}}}}}}}|r|s|dk(r|s |rer |||||||f$c}}}}}}}dZ'e jBjHe jBjKdeeddgdgddgddgddgDcgc]\}}}}}}|dk(r|s||||||fc}}}}}}dZ(e jBjHe jBjEe d e jBjEe d e jBjEejRd e jBjKdeeddgddgddgddgdgDcgc]'\}}}}}}|r|s|dk7r|s|r|s|dk(r|s||||||f)c}}}}}}dZ*e jBjHe jBjEejV d e jBjEe xsed e jBjKdeedgddgddgdgddgDcgc]$\}}}}}}|r|s|dk(r|ser|s |rer||||||f&c}}}}}}d!Z,e jBjEe d e jBjEed" e jBjEexred# e jBjEejRd d$Z-e jBjEejV d e jBjEe d% d&Z.e jBjEdd' e jBjEe d( d)Z/e jBjEdd* d+Z0e jBjHd,Z1e jBjHe jBjKd-eeddgddgddgDcgc]\}}}}|r|s||||fc}}}}d.Z2e jBjHd/Z3e jBjHe jBjKd0e4eeddgdgd1Z5e jBjEdd2 e jBjEe xsed3 d4Z6ycc}}}}}}wcc}}}}}}}wcc}}}}}}wcc}}}}}}wcc}}}}}}wcc}}}}w)5N)redirect_stderrredirect_stdout)product)Path)MODELSOURCE)YOLO) TASK2DATA TASK2MODELTASKS)ARM64IS_RASPBERRYPILINUXMACOSWINDOWSchecks) TORCH_1_11 TORCH_1_13 TORCH_2_1ctttjddd}t|tdy)zOTest YOLO model export to TorchScript format for compatibility and correctness. torchscriptF )formatoptimizeimgszrNr rexportrfiles P/var/www/html/ai-service/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>=2.1)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-@)rr3T ignore_errorsN)r r rr rrrename with_stemstemuuiduuid4rshutilrmtreetaskr&r1r2r3r5r s r!test_export_openvino_matrixrC-s  4 ! ( (   t_  ) DDz{{4>>TYYKq*GHIDJx%Wr"EJ MM$d+r#z/task, dynamic, int8, half, batch, simplify, nmsc tt|jdd||||||}t|tg|z|rdndt |j y)zKTest YOLO export to ONNX format with various configurations and parameters.r%r)rrr&r1r2r3simplifyr5r7rNr r rrrunlink)rBr&r1r2r3rEr5r s 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&r1r2r3r5r7rNrFrAs r!test_export_torchscript_matrixrKbsa  4 ! ( (BdUZ`c ) DDJx%Wr"=Jr#z(CoreML inference only supported on macOSz"CoreML export requires torch>=1.11z#CoreML not supported in Python 3.13z%task, dynamic, int8, half, nms, batchdetectc tt|jdd|||||}t|tg|zdt j |y)zHTest YOLO export to CoreML format with various parameter configurations.coremlrrJrNr r rrr?r@)rBr&r1r2r5r3r s r!test_export_coreml_matrixrPvs]&  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.tfliterrJrNrFrAs r!test_export_tflite_matrixrSs]$  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.rNTr)rr5rr+rNErrorzCoreML export produced errors: z%You will not be able to run predict()z!CoreML export has predict() error) ioStringIOrrr rrrrgetvalue)stdoutstderrr outputs 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.rRrr+rNrmodelr s r!test_export_tfliter`s/ KE <rs{ 7 88LK! !  M*HI!J!  M*HI+6= D%=4-$Au 6  1D'4uc$DJ$63 wdE3/ , J,*5@G D%=5'E7QFT5MDRW=@  ;D'4uh$DJ$63CPZ wdE8S9  +6= E4=5'E4=1a&4-6  1D'4uc "s wdE3/  I&PQ N+OPF))2WX+6= D%=4-$u PQs6    1D'4sE!cS# wdC/    YQR" 555>cd I b+6= E7T5MD%=1#e}6  1D'4uc$DJ$63EcWZcm wdE3/   e  N+OPG$EFEOe,STF))2WX fYUGQ f 555>cdI&op!qe!D1I&^_!`2!D!lm2n2 !! (/utUmdE]UVXYTZ'[ #D$e tT5!!! ,d754-RSQT3U.VW,X,D!noI&/ST!Up!g@(. DNs$'!Y2-'Y<3Z,Z+)ZZ%