Skip to content

Instantly share code, notes, and snippets.

@rpitonak

rpitonak/export_resize.py

Last active July 12, 2022 07:54
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/793258808a2d75dd6c5768d9140eb9a4 to your computer and use it in GitHub Desktop.
Save rpitonak/793258808a2d75dd6c5768d9140eb9a4 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
export_resize.py
graph = onnx.helper.make_graph([roi, scales, output_size, resizer_node, transpose_node], "resizer", [X], [Y])
print(graph)
# Create the model (ModelProto instance)
model_def = onnx.helper.make_model(g, producer_name="onnx-zaitra")
# these are optional, just to show what you can set here
model_def.opset_import[0].version = 12
model_def.ir_version = 7
model_def = onnx.shape_inference.infer_shapes(model_def)
onnx.checker.check_model(model_def)
onnx.save(model_def, "resizer.onnx")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment