Skip to content
This repository was archived by the owner on Feb 7, 2023. It is now read-only.

Commit 26bc5c4

Browse files
authored
Allow running convert.py file directly (#134)
1 parent 69fb6f0 commit 26bc5c4

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

onnx_coreml/bin/convert.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,3 +24,7 @@ def onnx_to_coreml(onnx_model, output): # type: (IO[str], str) -> None
2424
onnx_model_proto.ParseFromString(onnx_model.read())
2525
coreml_model = convert(onnx_model_proto)
2626
coreml_model.save(output)
27+
28+
29+
if __name__ == '__main__':
30+
onnx_to_coreml()

0 commit comments

Comments
 (0)