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>
This commit is contained in:
1 parent
7e21392561
commit
878d9c8d5b
1 file changed
+1
-1
@@ -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':
|
||||
|
||||
Reference in new issue
Block a user