Spaces:

xiang-wuu
/

yolov5

Runtime error

glenn-jocher commited on Apr 23, 2021

Commit

a2a514d

unverified ·

1 Parent(s): 28db237

Update export.py (#2909)

Files changed (1) hide show

models/export.py CHANGED Viewed

@@ -112,7 +112,7 @@ if __name__ == '__main__':
     try:
         import coremltools as ct
-        print(f'{prefix} starting export with coremltools {onnx.__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

     try:
         import coremltools as ct
+        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