为了促进将RealBasicVSR部署到其他平台上,请尝试将RealBasicvs的模型转换为ONNX格式,以记录此过程中遇到的问题和解决方案
我将以下代码添加到以下代码中以导出ONNX模型
通过将输入和输出设置为动态大小
上一个的最小示例是简单地加载图片和模型,然后对其进行处理,显示和保存,使用一些Numpy的方法来保持数据的数据和模型输入的数据一致
这是基于我的理解,即未传递test_mode字段,因此我直接更改了以下代码。最初是test_mode默认为false,我将其更改为true,不确定是否还有其他效果。只能说目前可用。
在发现了很长时间之后,我发现该模型实际上有两个输出。该模型被导出时应命名,然后选择第二个。第二个输出直接在上面的代码中选择
原始:https://juejin.cn/post/7101652566005514248