Fix float zeros format (#5491)

* Fix float zeros format

* 255 to integer
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-11-03 23:36:53 +01:00
1 parent 8a803f36d3
commit 5866646cc8
16 files changed
+36 -36

No files matched your search

+1 -1
View File
@@ -117,7 +117,7 @@ def export_coreml(model, im, file, prefix=colorstr('CoreML:')):
model.train() # CoreML exports should be placed in model.train() mode
ts = torch.jit.trace(model, im, strict=False) # TorchScript model
ct_model = ct.convert(ts, inputs=[ct.ImageType('image', shape=im.shape, scale=1 / 255.0, bias=[0, 0, 0])])
ct_model = ct.convert(ts, inputs=[ct.ImageType('image', shape=im.shape, scale=1 / 255, bias=[0, 0, 0])])
ct_model.save(f)
LOGGER.info(f'{prefix} export success, saved as {f} ({file_size(f):.1f} MB)')