onnx-community

/

rtdetr_r50vd

like 0

Patch script:

import onnx
def patch(model):
    node = model.graph.node
    idx = 0
    while idx < len(node):
        n = node[idx]
        if n.op_type == "AveragePool" and n.attribute[0].i != 0:
            n.attribute[0].i = 0
        idx += 1

    return model