a yfj@s(ddlZddlZddlmZddlmZddlZddlmZm Z ddl m Z ddl m Z mZmZddlmZmZmZmZmZddlmZmZd d Zd d Zejje d dddZejjejje d dejdddeeddgddgddgddgDddZ ejjejdeeddgdgdgddgddgddZ!ejjejdeedgdgdgddgddZ"ejjejje ddejje d dejjej#d!dejdd"deedgddgddgdgDd#d$Z$ejjejjej% d%dejje d&dejdd'deedgddgddgdgDd(d)Z&ejje d dejjed*dejjed+dejjej#d!dd,d-Z'ejjej% d%dejje d&dd.d/Z(ejjdd0dejje d1dd2d3Z)ejjdd4dd5d6Z*ejjd7d8Z+dS)9N)product)Path)MODELSOURCE)YOLO) TASK2DATA TASK2MODELTASKS)IS_RASPBERRYPILINUXMACOSWINDOWSchecks) TORCH_1_9 TORCH_1_13cCs(ttjdddd}t|tdddS)zRTest YOLO model exporting to TorchScript format for compatibility and correctness. torchscriptF )formatoptimizeimgszrNrrexportrfilerN/var/www/html/django/DPS/env/lib/python3.9/site-packages/tests/test_exports.pytest_export_torchscriptsrcCs(ttjdddd}t|tdddS)z8Test YOLO model export to ONNX format with dynamic axes.onnxTr)rdynamicrrNrrrrrtest_export_onnxsr zOpenVINO requires torch>=1.13)reasoncCs&ttjddd}t|tdddS)zGTest YOLO exports to OpenVINO format for model inference compatibility.openvinorrrrNrrrrrtest_export_openvino#sr$z task, dynamic, int8, half, batchcCs,g|]$\}}}}}|r|s|||||fqSrr.0taskrint8halfbatchrrr .s r+TFc Cstt|jdd||||t|d}trPt|}|||jdt }t|t g||rfdnddt j |ddd S) zRTest YOLO model exports to OpenVINO under various configuration matrix conditions.r"r)rrrr(r)r*data-@rT) ignore_errorsN)rrrrr rrename with_stemstemuuiduuid4rshutilrmtreer'rr(r)r*rrrrtest_export_openvino_matrix*s   r:z*task, dynamic, int8, half, batch, simplifyc CsNtt|jdd|||||d}t|tg||r6dnddt|dS)zLTest YOLO exports to ONNX format with various configurations and parameters.rr)rrrr(r)r*simplifyr0rNrrrrrunlink)r'rr(r)r*r;rrrrtest_export_onnx_matrixHs  r>cCsLtt|jdd||||d}t|tgd|r4dnddt|dS)zKTests YOLO model exports to TorchScript format under varied configurations.rrrrrr(r)r*r0rNr<r9rrrtest_export_torchscript_matrix[s rAz(CoreML inference only supported on macOSz+CoreML>=7.2 not supported with PyTorch<=1.8z#CoreML not supported in Python 3.12cCs,g|]$\}}}}}|r|s|||||fqSrrr%rrrr+qs cCsBtt|jdd||||d}t|tg|ddt|dS)zITest YOLO exports to CoreML format with various parameter configurations.coremlrr?rN)rrrrr7r8r9rrrtest_export_coreml_matrixks rCz#TFLite export requires Python>=3.10zGTest disabled as TF suffers from install conflicts on Windows and macOScCs,g|]$\}}}}}|r|s|||||fqSrrr%rrrr+s cCsDtt|jdd||||d}t|tg|ddt|dS)zMTest YOLO exports to TFLite format considering various export configurations.tfliterr?rNr<r9rrrtest_export_tflite_matrixs rEzCoreML not supported on Windowsz$CoreML not supported on Raspberry PicCs@tr(ttjddd}t|tddnttjdddddS)z=Test YOLO exports to CoreML format, optimized for macOS only.rBrr#rT)rZnmsrN)r rrrrrrrrtest_export_coremlsrFcCs*tt}|jddd}t|tdddS)zSTest YOLO exports to TFLite format under specific OS and Python version conditions.rDrr#rNrmodelrrrrtest_export_tflitesrIz Test disabledz6TF suffers from install conflicts on Windows and macOScCs*tt}|jddd}t|tdddS)z9Test YOLO exports to TensorFlow's Protobuf (*.pb) format.Zpbrr#rNrrGrrrtest_export_pbsrJzITest disabled as Paddle protobuf and ONNX protobuf requirements conflict.cCsttjddddS)zHTest YOLO exports to Paddle format, noting protobuf conflicts with ONNX.Zpaddlerr#N)rrrrrrrtest_export_paddlesrKcCs&ttjddd}t|tdddS)z!Test YOLO exports to NCNN format.Zncnnrr#rNrrrrrtest_export_ncnnsrL),r7r5 itertoolsrpathlibrZpytesttestsrrZ ultralyticsrZultralytics.cfgrrr Zultralytics.utilsr r r r rZultralytics.utils.torch_utilsrrrr markZskipifr$ZslowZ parametrizer:r>rAZIS_PYTHON_3_12rCZIS_PYTHON_MINIMUM_3_10rErFrIrJrKrLrrrrs    " "