Unverified 提交 a2a514de authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Update export.py (#2909)

上级 28db2376
......@@ -112,7 +112,7 @@ if __name__ == '__main__':
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
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论