From a9918fbf2b6648bbb4bac3e13f14217a54ee28a4 Mon Sep 17 00:00:00 2001 From: Laurentiu Diaconu Date: Wed, 8 Jul 2020 09:43:33 +0300 Subject: [PATCH] updated coreml conversion to do pixel scaling between 0.0 and 1.0 --- models/export.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/models/export.py b/models/export.py index c11c0a3..990c86e 100644 --- a/models/export.py +++ b/models/export.py @@ -61,7 +61,8 @@ if __name__ == '__main__': 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)