cszn / SCUNet

Practical Blind Denoising via Swin-Conv-UNet and Data Synthesis (Machine Intelligence Research 2023)
https://link.springer.com/article/10.1007/s11633-023-1466-0
Apache License 2.0
637 stars 63 forks source link

模型‘scunet_color_real_gan.pth’ 导出为onnx,动态尺寸推理不成功 #14

Open SophiaDz001 opened 2 years ago

SophiaDz001 commented 2 years ago

pth转为onnx,由于mul操作不支持矩阵于矩阵相乘的动态尺寸,不知道作者是否有尝试过导出动态尺寸的onnx模型成功推理嘛?

lhehejunl commented 1 year ago

pth转为onnx,由于mul操作不支持矩阵于矩阵相乘的动态尺寸,不知道作者是否有尝试过导出动态尺寸的onnx模型成功推理嘛?

我也在导出onnx,导出后用的opencv加载失败,请问您固定尺寸导出的onnx模型推理成功了吗?