Update export.py (ultralytics#11638)

* Update export.py Signed-off-by: hackerrajeshkumar <[email protected]> * Update export.py Signed-off-by: Glenn Jocher <[email protected]> --------- Signed-off-by: hackerrajeshkumar <[email protected]> Signed-off-by: Glenn Jocher <[email protected]> Co-authored-by: Glenn Jocher <[email protected]>
ZQFORWARD · 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':