Matlab基于连续小波变换（CWT）批量生成时频图

Matlab基于连续小波变换CWT将一维信号批量生成时频图的源 此示例中原始信号data是30*1280的格式一共30条信号信号长度为1280。 最终生成30张时频图。 生成的图像可用于后续的深度学习分类或其他处理。 附详细的说明文档。 程序工作如下 1、加载信号定义好采样频率。 2、画出一个信号的时频图这个是为了看一个信号时频图长什么样以及方便放在论文里。 3、构建结构体存放信号和类别这里假设所有信号均是Normal类别。 4、批量生成时频图存放于文件夹中。 需要Matlab2020及以上版本。 在改为自己的信号时只需要照样子整理好数据的格式然后更改采样频率和存放时频图的路径即可。 程序不用大幅修改。 主程序main.m中代码仅30多行注释详细方便看懂。 两个子函数helperCreateECGDirectories.m和helperCreateRGBfromTF.m分别为创建空文件夹和批量生成时频图的函数替换信号时只需要修改其中的采样频率即可。在信号处理和深度学习的交叉领域中时频图作为一种强大的工具能够直观地展示信号在时间和频率域的联合分布为后续的分类等任务提供有效的特征表示。今天就来分享一下如何用Matlab基于连续小波变换CWT将一维信号批量生成时频图。数据格式与目标我们的原始信号data是30*1280的格式即一共30条信号每条信号长度为1280。最终要生成30张时频图这些图像可用于后续的深度学习分类或其他处理。程序工作流程1. 加载信号与定义采样频率在Matlab中加载信号通常可以使用load函数如果信号存储在.mat文件中。假设我们的信号存储在data.mat文件中代码如下load(data.mat); % 加载信号数据 fs 1000; % 定义采样频率这里假设为1000Hz实际使用需根据信号情况调整这里定义的采样频率fs非常关键它决定了信号在时间维度上的离散化程度对后续时频图的准确性有很大影响。2. 画出单个信号的时频图这一步主要是为了预览时频图的样子方便在论文等文档中展示。以第一条信号为例使用连续小波变换来生成时频图代码如下signal data(1,:); % 选取第一条信号 wname db4; % 选择小波基这里使用db4小波 scales 1:128; % 定义尺度范围 [cfs,f] cwt(signal,scales,wname,1/fs); % 进行连续小波变换 figure; tfrscalogram(cfs,f,1/fs,yaxis,linear); % 绘制时频图 title(Single Signal Time - Frequency Plot);在这段代码中首先选择了第一条信号然后确定了小波基wname和尺度范围scales通过cwt函数进行连续小波变换得到系数cfs和频率f最后使用tfrscalogram函数绘制时频图。3. 构建结构体存放信号和类别这里假设所有信号均是Normal类别构建结构体的代码如下dataStruct struct(signal, {}, class, {}); for i 1:size(data,1) dataStruct(i).signal data(i,:); dataStruct(i).class Normal; end通过循环将每条信号及其类别存入结构体dataStruct中方便后续批量处理。4. 批量生成时频图并存放于文件夹主程序main.m代码量仅30多行且注释详细。这里我们来看两个关键的子函数。helperCreateECGDirectories.m这个函数用于创建存放时频图的空文件夹。代码如下function helperCreateECGDirectories(tfDir) if ~isfolder(tfDir) mkdir(tfDir); end end函数接收一个路径参数tfDir如果该路径不存在则使用mkdir函数创建文件夹。helperCreateRGBfromTF.m此函数用于批量生成时频图。代码如下function helperCreateRGBfromTF(dataStruct,fs,tfDir) wname db4; scales 1:128; for i 1:length(dataStruct) signal dataStruct(i).signal; [cfs,f] cwt(signal,scales,wname,1/fs); tfImage tfrscalogram(cfs,f,1/fs,yaxis,linear); tfImage mat2gray(tfImage); tfImage repmat(tfImage,[1 1 3]); imwrite(tfImage,fullfile(tfDir,[tfImage_ num2str(i) .png])); end end在这个函数中首先确定小波基和尺度范围然后遍历结构体中的每条信号进行连续小波变换得到时频图数据tfImage将其转换为灰度图像并扩展为RGB格式最后使用imwrite函数将图像保存到指定文件夹中。使用说明此程序需要Matlab2020及以上版本。当你要改为自己的信号时只需要照样子整理好数据的格式使其成为类似n*m的矩阵形式n为信号数量m为每条信号长度然后更改采样频率和存放时频图的路径即可程序不用大幅修改。Matlab基于连续小波变换CWT将一维信号批量生成时频图的源 此示例中原始信号data是30*1280的格式一共30条信号信号长度为1280。 最终生成30张时频图。 生成的图像可用于后续的深度学习分类或其他处理。 附详细的说明文档。 程序工作如下 1、加载信号定义好采样频率。 2、画出一个信号的时频图这个是为了看一个信号时频图长什么样以及方便放在论文里。 3、构建结构体存放信号和类别这里假设所有信号均是Normal类别。 4、批量生成时频图存放于文件夹中。 需要Matlab2020及以上版本。 在改为自己的信号时只需要照样子整理好数据的格式然后更改采样频率和存放时频图的路径即可。 程序不用大幅修改。 主程序main.m中代码仅30多行注释详细方便看懂。 两个子函数helperCreateECGDirectories.m和helperCreateRGBfromTF.m分别为创建空文件夹和批量生成时频图的函数替换信号时只需要修改其中的采样频率即可。通过以上步骤我们就可以利用Matlab基于连续小波变换实现一维信号的批量时频图生成为后续的深度学习等任务提供有力的数据支持。希望这篇博文能对大家在信号处理相关工作中有所帮助