下载项目
WNO在github上面的项目地址如下:
https://github.com/csccm-iitd/WNO/tree/main
下载下来后,里面的数据集需要用matlab代码生成,也可以到里面提到的google云盘里面下载数据集
安装环境
然后需要安装环境
运行代码会报错如下,提示安装下面的三个环境
Wavelet convolution requires <Pytorch Wavelets>, <PyWavelets>, <Pytorch Wavelet Toolbox> \n \For Pytorch Wavelet Toolbox: $ pip install ptwt \n \For PyWavelets: $ conda install pywavelets \n \For Pytorch Wavelets: $ git clone https://github.com/fbcotter/pytorch_wavelets \n \$ cd pytorch_wavelets \n \$ pip install .')
注意:pytorch_wavelets这个文件最好和要运行的WNO相关的python文件在同一个目录,不然可能会报错。
报错内容是:找不到DWT1D这个模型
将pytorch_wavelets放到代码同目录然后再pip install .安装后即可解决这个问题