Skip to content

Instantly share code, notes, and snippets.

@rpitonak

rpitonak/nodes.py

Created July 12, 2022 07:51
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/3bbb2196ac0007ffd60ed5c54bd92a40 to your computer and use it in GitHub Desktop.
Save rpitonak/3bbb2196ac0007ffd60ed5c54bd92a40 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
nodes.py
resizer_node = onnx.helper.make_node("Resize", inputs=[model_input_name, "roi", "scales", "output_size"], outputs=[resizer_node_output_name])
transpose_node = onnx.helper.make_node("Transpose", inputs=[resizer_node_output_name], outputs=[model_output_name], perm=[0,3,1,2])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment