1

pytorch 中没有指定输入的大小。只需调整内核大小以进行输出。WinMLDashboard 显示图像输入的宽度和高度。这怎么可能?

4

1 回答 1

1

你的意思是当你将网络从pytorch序列化到onnx时?因为当您从 pytorch 导出时,您需要根据文档定义输入的大小

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)
于 2019-07-15T10:25:39.960 回答