Update export.py (ultralytics#2909)

Lechtr · Apr 23, 2021 · a2a514d · a2a514d
1 parent 28db237
commit a2a514d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/export.py b/models/export.py
@@ -112,7 +112,7 @@
     try:
         import coremltools as ct
 
-        print(f'{prefix} starting export with coremltools {onnx.__version__}...')
+        print(f'{prefix} starting export with coremltools {ct.__version__}...')
         # convert model from torchscript and apply pixel scaling as per detect.py
         model = ct.convert(ts, inputs=[ct.ImageType(name='image', shape=img.shape, scale=1 / 255.0, bias=[0, 0, 0])])
         f = opt.weights.replace('.pt', '.mlmodel')  # filename