ONNX Surgery example - Replacing `GlobalAveragePool` with `AveragePool`

onnx-surgery.py

import onnx

model = onnx.load("model.onnx")
graph = model.graph

# Identify relevant node
idx = -1
for i, n in enumerate(model.graph.node):
    if n.name == "GlobalAveragePool":
        idx = i
        
node  = model.graph.node[idx]
print(node)
new_node = onnx.helper.make_node("AveragePool", node.input, node.output, kernel_shape=[12,4], pads=[0,0,0,0], strides=[12,4])

graph.node.remove(node)
graph.node.insert(idx, new_node)
onnx.checker.check_model(model)

onnx.save(model, "model-surgery.onnx")