diff --git a/export.py b/export.py index 815a27cb0d43..9399f0bbd221 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':