matlab填充图像算法代码CNN_频谱图_算法
一种使用卷积神经网络
(CNN)
从原始
EEG
数据中对频谱图进行分类的方法
SpectrogramClassificationAlgorithm.ipynb:使用来自训练和有效文件夹的数据训练模型,对来自测试文件夹的数据进行测试
Use_pretrained_model.ipynb:使用来自训练模型的模型权重,测试来自测试文件夹的新测试数据
fastai
文件夹中的代码来自
Jeremy
Howard
的
fastai
0.7
版:
数据结构
要运行任一笔记本,您必须具有以下结构的数据文件夹:
数据/
├──火车/
├──
Yes
├──
No
├──
有效/
├──
Yes
├──
No
├──测试/
对于
SpectrogramClassificationAlgorithm.ipynb:Yes
和
No
子文件夹包含我们训练模型的正面和负面案例图像。
测试文件夹包含我们测试模型的未分类图像。
对于
Use_pretrained_model.ipynb:测试文件夹包含要由我们的预训练模型
(saved_model.pkl)
评估的
2022-04-15 01:53:45
7.77MB
系统开源
1