首页 > 解决方案 > 将模型从matlab导出到onnx时,如何使模型可以接受任何大小的输入?

问题描述

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

标签: c#matlabdeep-learningml.netonnx

解决方案


推荐阅读