如何知道 onnx 中模型的输入大小？
how to know model's input size in onnx?
pytorch中没有指定输入的大小。只需调整内核大小即可生成输出。 WinMLDashboard 显示图像输入的宽度和高度。这怎么可能？
你是说当你把网络从pytorch序列化到onnx的时候？因为当您从 pytorch 导出时，您需要根据 documentation
定义输入的大小
dummy_input = torch.randn(10, 3, 224, 224, device='cuda')
model = torchvision.models.alexnet(pretrained=True).cuda()
input_names = [ "actual_input_1" ]
output_names = [ "output1" ]
torch.onnx.export(model, dummy_input, "alexnet.onnx", verbose=True,
input_names=input_names, output_names=output_names)
pytorch中没有指定输入的大小。只需调整内核大小即可生成输出。 WinMLDashboard 显示图像输入的宽度和高度。这怎么可能？
你是说当你把网络从pytorch序列化到onnx的时候？因为当您从 pytorch 导出时，您需要根据 documentation
定义输入的大小dummy_input = torch.randn(10, 3, 224, 224, device='cuda')
model = torchvision.models.alexnet(pretrained=True).cuda()
input_names = [ "actual_input_1" ]
output_names = [ "output1" ]
torch.onnx.export(model, dummy_input, "alexnet.onnx", verbose=True,
input_names=input_names, output_names=output_names)