Ankle/yolov5_ZWgra
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Update export.py (#11638)

Browse Source 
* 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:
hackerrajeshkumarandGlenn Jocher authored and GitHub committed 2023-06-17 20:51:50 +02:00
1 parent 7e21392561
commit 878d9c8d5b
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
export.py
+1 -1
View File
@@ -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 batch_size, ch, h, w = list(im.shape) # BCHW
t = time.time() t = time.time()
# Output shapes # YOLOv5 Output shapes
spec = model.get_spec() spec = model.get_spec()
out0, out1 = iter(spec.description.output) out0, out1 = iter(spec.description.output)
if platform.system() == 'Darwin': if platform.system() == 'Darwin':