Update export.py

Signed-off-by: hackerrajeshkumar <120269593+hackerrajeshkumar@users.noreply.github.com>
ultralytics · Jun 1, 2023 · 714dc24 · 714dc24
1 parent 5eb7f7d
commit 714dc24
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/export.py b/export.py
@@ -536,7 +536,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':