从Matlab向Onnx导出模型时,如何使模型可以接受任意大小的输入?

时间:2020-09-25 06:55:15

标签: c# matlab deep-learning ml.net onnx

我得到了一个模型(带有Matlab的火车),它是FCN(完全卷积网络)。输入是一幅图像,输出是该图像的噪声,因此我们的目标是(输入-输出)这是一个去噪图像。由于FCN结构,从理论上讲,它应该采用任何大小的输入。但是我将模型输入大小设置为50x50,以加快训练速度。然后,我将模型导出为ONNX(开放式神经网络交换)格式,并尝试使用Ml.Net推断C#。然后,我发现模型输入大小必须与训练(50x50)相同。我希望模型可以推断任何大小的输入。我该如何解决这个问题?正如我在Keras中所知,可以将输入大小设置为“无”以获取任何大小的输入。但是我不知道如何在Matlab中做到这一点。有人有什么想法吗?我使用下面的代码。 Matlab denoisedImage

0 个答案:

没有答案