Obtain ONNX Input Size
作者：XD / 发表： 2021年9月25日 08:18 / 更新： 2021年9月25日 08:18 / 编程笔记 / 阅读量：3312
Obtain ONNX Input Size
import onnx
path = "onnx_model.onnx"
model = onnx.load(path)
input_shape = model.graph.input[0].type.tensor_type.shape.dim
height = int(str(input_shape[2]).strip().split(': ')[-1])
width = int(str(input_shape[3]).strip().split(': ')[-1])
