From 878d9c8d5b21253ee3a086b69a94fbbf55e56088 Mon Sep 17 00:00:00 2001 From: hackerrajeshkumar <120269593+hackerrajeshkumar@users.noreply.github.com> Date: Sun, 18 Jun 2023 00:21:50 +0530 Subject: [PATCH] Update export.py (#11638) * Update export.py Signed-off-by: hackerrajeshkumar <120269593+hackerrajeshkumar@users.noreply.github.com> * Update export.py Signed-off-by: Glenn Jocher --------- Signed-off-by: hackerrajeshkumar <120269593+hackerrajeshkumar@users.noreply.github.com> Signed-off-by: Glenn Jocher Co-authored-by: Glenn Jocher --- export.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/export.py b/export.py index 815a27c..9399f0b 100644 --- a/export.py +++ b/export.py @@ -581,7 +581,7 @@ def pipeline_coreml(model, im, file, names, y, prefix=colorstr('CoreML Pipeline: batch_size, ch, h, w = list(im.shape) # BCHW t = time.time() - # Output shapes + # YOLOv5 Output shapes spec = model.get_spec() out0, out1 = iter(spec.description.output) if platform.system() == 'Darwin':