python pytorch fft_用Pytorch实现FFT

我尝试使用Pytorch中提供的conv1d函数来实现FFT。在

产生人工信号import numpy as np

import torch

from torch.autograd import Variable

from torch.nn.functional import conv1d

from scipy import fft, fftpack

import matplotlib.pyplot as plt

%matplotlib inline

# Creating filters

d = 4096 # size of windows

def create_filters(d):

x = np.arange(0, d, 1)

wsin = np.empty((d,1,d), dtype=np.float32)

wcos = np.empty((d,1,d), dtype=np.float32)

window_mask = 1.0-1.0*np.cos(x)

for ind in range(d):

wsin[ind,0,:] = np.sin(2*np.pi*((ind+1)/d)*x)

wcos[ind,0,:] = np.cos(2*np.pi*((ind+1)/d)*x)

return wsin,wcos

wsin, wcos = create_filters(d)

wsin_var = Variable(torch.from_numpy(wsin), requires_grad=False)

wcos_var = Variable(torch.from_numpy(wcos),requires_grad=False)

# Creating signal

t = np.linspace(0,1,4096)

x = np.sin(2*np.pi*100*t)+np.sin(2*np.pi*200*t)+np.random.normal(scale=5,size=(4096))

plt.plot(x)

带Pythorch的FFT

^{pr2}$

带Scipy的FFTfig = plt.figure(figsize=(20,5))

plt.plot(np.abs(fft(x).reshape(-1))[:500])

我的问题

如您所见，这两个输出在峰值特性方面非常相似。这意味着我的实现并不是完全错误的。

然而，也有一些微妙之处，比如频谱的尺度和信噪比。我不知道这里缺了什么来得到完全相同的结果。在

python pytorch fft_用Pytorch实现FFT相关推荐

python pytorch fft_看PyTorch源代码的心路历程
1. 起因曾经碰到过别人的模型prelu在内部的推理引擎算出的结果与其在原始框架PyTorch中不一致的情况,虽然理论上大家实现的都是一个算法,但是从参数上看,因为经过了模型转换,中间做了一些调整. ...
【深度学习】基于Torch的Python开源机器学习库PyTorch卷积神经网络
[深度学习]基于Torch的Python开源机器学习库PyTorch卷积神经网络文章目录 1 CNN概述 2 PyTorch实现步骤2.1 加载数据2.2 CNN模型2.3 训练2.4 可视化训练 ...
【深度学习】基于Torch的Python开源机器学习库PyTorch回归
[深度学习]基于Torch的Python开源机器学习库PyTorch回归文章目录1 torch.autograd 2 torch.nn.functional 3 详细的回归DEMO3.1 DATAS ...
【深度学习】基于Torch的Python开源机器学习库PyTorch概述
[深度学习]基于Torch的Python开源机器学习库PyTorch概述文章目录 1 PyTorch简介 2 环境搭建 3 Hello world!3.1 Tensors (张量)3.2 操作 4 ...
图像迁移风格保存模型_图像风格迁移也有框架了：使用Python编写，与PyTorch完美兼容，外行也能用...
原标题:图像风格迁移也有框架了:使用Python编写,与PyTorch完美兼容,外行也能用选自Medium 作者:Philip Meier 机器之心编译编辑:陈萍易于使用的神经风格迁移框架 py ...
利用python安装opencv_科学网—Anaconda Python PyCharm PyQT5 OpenCV PyTorch TF2.0 安装指南 - 张重生的博文...
Anaconda Python PyCharm PyQT5 OpenCV PyTorch TF2.0 安装指南与资料汇总 (用Anaconda配置Python集成开发环境,含Python3, PyQT ...
python pytorch语音识别_PyTorch通过ASR实现语音到文本端的模型以及pytorch语音识别(speech) - pytorch中文网...
ASR,英文的全称是Automated Speech Recognition,即自动语音识别技术,它是一种将人的语音转换为文本的技术.今天我们主要了解pytorch实现语音到文本的端到端模型. spe ...
Python：机器学习模块PyTorch【上】
点击访问:PyTorch中文API应用具体代码地址自动求导机制本说明将概述Autograd如何工作并记录操作.了解这些并不是绝对必要的,但我们建议您熟悉它,因为它将帮助您编写更高效,更简洁的程序, ...
python绘制图形沙漏_pytorch-pose一个用于二维人体姿势估计的PyTorch工具包。 - pytorch中文网...
pytorch-pose PyTorch-Pose是2D单人姿态估计的一般流水线的PyTorch实现.其目的是为最流行的人体姿态数据库(如MPII人体姿态,LSP和FLIC)提供训练/推理/评估的接口 ...

python pytorch fft_用Pytorch实现FFT

python pytorch fft_用Pytorch实现FFT相关推荐

最新文章

热门文章