File size: 248 Bytes
eef0ae4
 
 
 
 
 
1
2
3
4
5
6
7
import onnx

model = onnx.load("onnx/frida-onnx/FRIDA.onnx")
for inp in model.graph.input:
    shape = [dim.dim_value if dim.dim_value > 0 else str(dim.dim_param) for dim in inp.type.tensor_type.shape.dim]
    print(f"Input '{inp.name}': {shape}")