diff --git a/models/export.py b/models/export.py index c11c0a391197..990c86e92313 100644 --- a/models/export.py +++ b/models/export.py @@ -61,7 +61,8 @@ import coremltools as ct print('\nStarting CoreML export with coremltools %s...' % ct.__version__) - model = ct.convert(ts, inputs=[ct.ImageType(name='images', shape=img.shape)]) # convert + # convert model from torchscript and apply pixel scaling as per detect.py + model = ct.convert(ts, inputs=[ct.ImageType(name='images', shape=img.shape, scale=1/255.0, bias=[0, 0, 0])]) f = opt.weights.replace('.pt', '.mlmodel') # filename model.save(f) print('CoreML export success, saved as %s' % f)