pytorch的DCT(离散余弦变换)
该库根据pytorch中的内置FFT操作实现了DCT,从而可以在CPU和GPU上通过它进行反向传播。 有关DCT和此处使用的算法的更多信息,请参阅和的论文。 这篇也可能会有所帮助。
当前实现了以下内容:
一维DCT-I及其逆数(按比例缩放的DCT-I)
一维DCT-II及其逆数(按比例缩放的DCT-III)
2-D DCT-II及其逆数(按比例缩放的DCT-III)
3-D DCT-II及其逆数(按比例缩放的DCT-III)
安装
pip install torch-dct
要求torch>=0.4.1 (较低的版本可能还可以,但我尚未对其进行测试)。
您可以通过获取源代码运行测试并运行pytest 。 要运行测试,您还需要安装scipy 。
用法
import torch
import torch_dct as dct
x = tor
1