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 <glenn.jocher@ultralytics.com> --------- Signed-off-by: hackerrajeshkumar <120269593+hackerrajeshkumar@users.noreply.github.com> Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
ultralytics · Jun 17, 2023 · 878d9c8 · 878d9c8
1 parent 7e21392
commit 878d9c8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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':