h[+|SSKrSSKrSSKrSSKJrJr SSKJr SSKJ r SSK r SSK J r J r SSKJr SSKJrJrJr SSKJrJrJrJrJrJr SS KJrJrJr S rS r \ RBRE\(+S S 9S5r#\ RBRH\ RBRE\(+S S 9\ RBRKS\"\SS/SS/SS/SS/SS/5VVVVVVs/sH+upp#pEU(a U(aMUS:Xa U(aM&XX#XE4PM- snnnnnn5S555r&\ RBRH\ RBRKS\"\SS/S/S/SS/SS/SS/5VVVVVVVs/sH=upp#pFnU(a U(aMUS:Xa U(aM'U(a \(dM7XX#XFU4PM? snnnnnnn5S55r'\ RBRH\ RBRKS\"\SS/S/SS/SS/SS/5VVVVVVs/sHupp#pEUS:Xa U(aMXX#XE4PM snnnnnn5S55r(\ RBRH\ RBRE\(+SS 9\ RBRE\(+SS 9\ RBRE\RRSS 9\ RBRKS\"\SS/SS/SS/SS/S/5VVVVVVs/sHJupp#pTU(a U(aMUS:wa U(aM&U(a U(aM6US:Xa U(aMEXX#XT4PML snnnnnn5S55555r*\ RBRH\ RBRE\RV(+SS 9\ RBRE\(+=(d \S S 9\ RBRKS\"\S/SS/SS/S/SS/5VVVVVVs/sHKupp#pEU(a U(aMUS:Xa U(aM&\(a U(aM6U(a \(dMFXX#XE4PMM snnnnnn5S!5555r,\ RBRE\(+SS 9\ RBRE\S"S 9\ RBRE\=(a \S#S 9\ RBRE\RRSS 9S$5555r-\ RBRE\RV(+SS 9\ RBRE\(+S%S 9S&55r.\ RBRESS'S 9\ RBRE\(+S(S 9S)55r/\ RBRESS*S 9S+5r0\ RBRHS,5r1\ RBRH\ RBRKS-\"\SS/SS/SS/5VVVVs/sHupp4U(a U(aMXX44PM snnnn5S.55r2\ RBRHS/5r3\ RBRH\ RBRKS0\4"\"\SS/S/555S155r5\ RBRESS2S 9\ RBRE\(+=(d \S3S 9S455r6gs snnnnnnfs snnnnnnnfs snnnnnnfs snnnnnnfs snnnnnnfs snnnnf)5N)redirect_stderrredirect_stdout)product)Path)MODELSOURCE)YOLO) TASK2DATA TASK2MODELTASKS)ARM64IS_RASPBERRYPILINUXMACOSWINDOWSchecks) TORCH_1_11 TORCH_1_13 TORCH_2_1ch[[5RSSSS9n[U5"[SS9 g)zOTest YOLO model export to TorchScript format for compatibility and correctness. torchscriptF )formatoptimizeimgszrNr rexportrfiles L/home/james-whalen/.local/lib/python3.13/site-packages/tests/test_exports.pytest_export_torchscriptr"s+ ;  ]U"  MDJvR ch[[5RSSSS9n[U5"[SS9 g)z8Test YOLO model export to ONNX format with dynamic axes.onnxTr)rdynamicrrNrrs r!test_export_onnxr' s+ ;  VT  DDJvR r#zOpenVINO requires torch>=2.1)reasoncf[[5RSSS9n[U5"[SS9 g)zFTest YOLO export to OpenVINO format for model inference compatibility.openvinorrrrNrrs r!test_export_openvinor,&s+ ;  Zr  :DJvR r#z%task, dynamic, int8, half, batch, nmsTFclassifyc |[[U5RSSUUUU[UUS9n[(aM[ U5nUR URURS[R"5355n[U5"[/U-U(aSOSUS9 [R"USS9 g ) 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_  ) DwDz{{4>>TYYKq*GHIJx%Wr"EJ MM$d+r#z/task, dynamic, int8, half, batch, simplify, nmsc [[U5RSSXX4XVS9n[U5"[/U-U(aSOSS9 [ U5R 5 g)zKTest YOLO export to ONNX format with various configurations and parameters.r%r)rrr&r1r2r3simplifyr5r7rNr r rrrunlink)rBr&r1r2r3rEr5r s r!test_export_onnx_matrixrHNsZ  4 ! ( (R$^f ) D Jx%Wr"=Jr#c [[U5RSSXX4US9n[U5"[/U-U(aSOSS9 [ U5R 5 g)zITest YOLO model export to TorchScript format under varied configurations.rrrrr&r1r2r3r5r7rNrFrAs r!test_export_torchscript_matrixrKbsZ  4 ! ( (B`c ) D Jx%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 [[U5RSSUUUUUS9n[U5"[/U-SS9 [R "U5 g)zHTest YOLO export to CoreML format with various parameter configurations.coremlrrJrNr r rrr?r@)rBr&r1r2r5r3r s r!test_export_coreml_matrixrPvs[&  4 ! ( (    ) D Jx%r* MM$r#z#TFLite export requires Python>=3.10zUTest disabled as TF suffers from install conflicts on Windows, macOS and Raspberry Pic [[U5RSSXX4US9n[U5"[/U-SS9 [ U5R 5 g)zLTest YOLO export to TFLite format considering various export configurations.tfliterrJrNrFrAs r!test_export_tflite_matrixrSsV$  4 ! ( (r7D[^ ) D Jx%r*Jr#zCoreML not supported on Windowsz%CoreML not supported on aarch64 Linuxc[R"5[R"5p[U5 [U5 [ [ 5R SSSS9 [(a1[ [ 5R SSS9n[ U5"[SS9 SSS5 SSS5 UR5UR5-nSU;d S U35eS U;dS 5eg!,(df  NU=f!,(df  N^=f) 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[[]BKKMF  /&"9 U (B? 5;%%XR%@D JvR ( #: __ !2 2F & L$CF8"LL 2& @eBee @#:"9 s$ C<AC+C<+ C9 5C<< D zGTest disabled as TF suffers from install conflicts on Windows and macOScj[[5nURSSS9n[U5"[SS9 g)zRTest YOLO export to TFLite format under specific OS and Python version conditions.rRrr+rNrmodelr s r!test_export_tfliter`s- KE <[[5RSSS9 g)zGTest YOLO export to Paddle format, noting protobuf conflicts with ONNX.paddlerr+N)r rrr#r!test_export_paddlergs Khb1r#cf[[5RSSS9n[U5"[SS9 g)zaTest YOLO export to MNN format (WARNING: MNN test must precede NCNN test or CI error on Windows).mnnrr+rNrrs r!test_export_mnnrjs+ ;  U"  5DJvR r#ztask, int8, half, batchc[[U5RSSXUS9n[U5"[/U-SS9 [ U5R 5 g)zITest YOLO export to MNN format considering various export configurations.rir)rrr1r2r3rNrF)rBr1r2r3r s r!test_export_mnn_matrixrlsM  4 ! ( (Rd]b ( cDJx%r*Jr#cf[[5RSSS9n[U5"[SS9 g)z Test YOLO export to NCNN format.ncnnrr+rNrrs r!test_export_ncnnros+ ;  V2  6DJvR r#ztask, half, batchc[[U5RSSXS9n[U5"[/U-SS9 [R "USS9 g)zJTest YOLO export to NCNN format considering various export configurations.rnr)rrr2r3rTr8NrO)rBr2r3r s r!test_export_ncnn_matrixrqsH  4 ! ( (bt ( YDJx%r* MM$d+r#zKTest disabled as keras and tensorflow version conflicts with TFlite export.z"Skipping test on Windows and Macoscb[S5nURSSS9n[U5"[SS9 g)zTest YOLO export to IMX format.z yolov8n.ptimxrr+rN)r rrr^s r!test_export_imxrts.  E <rs 7 88LK! !  M*HI!J!  M*HI+6= D%=4-$Au 6 6 1D4u$ 0%)J$63 0E/6  , J,*5@G D%=5'E7QFT5MDRW=@ @ ;D4u$ :%)J$63 :EHPZ :ES9@   +6= E4=5'E4=1a&4-6 6 1D4u "s 0E/6   I&PQ N+OPF))2WX+6= D%=4-$u PQs6   6 1D4s 0 !c 0 S 0# 0C/6     YQR" 555>cd I b+6= E7T5MD%=1#e}6 6 1D4u$ 0%)J$63 0EJc 0X[cm 0E/6    e  N+OPG$EFEOe,STF))2WX fYUGQ f 555>cdI&op!qe!D1I&^_!`2!D!lm2n2 !! (/utUmdE]UVXYTZ'['[ #D "T!'[!! ,d754-RSQT3U.VW,X,D!noI&/ST!Up!g@(. DNs~)\ \ \\  \ * \ : \ #\< \\" \"+ \"; \" \" \,% \,4 \, \, \,\63\6