Skip to content

Instantly share code, notes, and snippets.

@rpitonak

rpitonak/merge_onnx.py

Last active June 10, 2022 13:00
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save rpitonak/b7e330511baab59a745f032f76e09bc9 to your computer and use it in GitHub Desktop.
Save rpitonak/b7e330511baab59a745f032f76e09bc9 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
merge_onnx.py
import onnx
# !wget https://github.com/onnx/models/raw/main/vision/classification/squeezenet/model/squeezenet1.0-12.onnx
model1 = onnx.load('resizer.onnx')
model2 = onnx.load('squeezenet1.0-12.onnx')
combined_model = onnx.compose.merge_models(
model1, model2,
io_map=[('Y', 'data_0')],
outputs=['softmaxout_1']
)
onnx.save(combined_model, "combined_dynamic_input.onnx")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment