在pytorch环境下,使用CNN模型对数据集MNIST进行深度特征提取,结合ELM进行图像分类的模型(分别用浅层特征,深层特征和深浅两个一起),得到准确率和训练时间

时间: 2024-05-05 08:23:00 浏览: 120
下面是使用PyTorch实现深度特征提取和ELM分类的代码,分别使用浅层特征、深层特征和深浅两个特征进行分类,最终输出训练时间和准确率。 ``` import torch import torch.nn as nn import torch.optim as optim import torchvision.datasets as datasets import torchvision.transforms as transforms import numpy as np from sklearn.preprocessing import StandardScaler from sklearn.metrics import accuracy_score from elm import ELMClassifier # 定义CNN模型 class CNN(nn.Module): def __init__(self): super(CNN, self).__init__() self.conv1 = nn.Conv2d(1, 32, kernel_size=5) self.conv2 = nn.Conv2d(32, 64, kernel_size=5) self.fc1 = nn.Linear(1024, 128) def forward(self, x): x = nn.functional.relu(self.conv1(x)) x = nn.functional.max_pool2d(x, 2) x = nn.functional.relu(self.conv2(x)) x = nn.functional.max_pool2d(x, 2) x = x.view(-1, 1024) x = nn.functional.relu(self.fc1(x)) return x # 加载数据集 train_dataset = datasets.MNIST(root='./data', train=True, transform=transforms.ToTensor(), download=True) test_dataset = datasets.MNIST(root='./data', train=False, transform=transforms.ToTensor()) # 定义超参数 batch_size = 256 lr = 0.001 num_epochs = 5 # 定义浅层特征和深层特征的ELM分类器 shallow_elm = ELMClassifier(hidden_layer_size=128) deep_elm = ELMClassifier(hidden_layer_size=128) # 定义深浅两个特征的ELM分类器 mixed_elm = ELMClassifier(hidden_layer_size=256) # 定义训练函数 def train(model, train_loader, optimizer, criterion): model.train() for batch_idx, (data, target) in enumerate(train_loader): optimizer.zero_grad() output = model(data) loss = criterion(output, target) loss.backward() optimizer.step() # 定义测试函数 def test(model, test_loader): model.eval() with torch.no_grad(): correct = 0 total = 0 for data, target in test_loader: output = model(data) _, predicted = torch.max(output.data, 1) total += target.size(0) correct += (predicted == target).sum().item() accuracy = 100 * correct / total return accuracy # 训练浅层特征的ELM分类器 train_loader = torch.utils.data.DataLoader(dataset=train_dataset, batch_size=batch_size, shuffle=True) test_loader = torch.utils.data.DataLoader(dataset=test_dataset, batch_size=batch_size, shuffle=False) model = CNN() criterion = nn.CrossEntropyLoss() optimizer = optim.Adam(model.parameters(), lr=lr) for epoch in range(num_epochs): train(model, train_loader, optimizer, criterion) shallow_train_features = [] shallow_train_labels = [] for data, target in train_loader: features = model(data) shallow_train_features.append(features) shallow_train_labels.append(target) shallow_train_features = torch.cat(shallow_train_features, dim=0).numpy() shallow_train_labels = torch.cat(shallow_train_labels, dim=0).numpy() scaler = StandardScaler() shallow_train_features = scaler.fit_transform(shallow_train_features) shallow_elm.fit(shallow_train_features, shallow_train_labels) shallow_test_features = [] shallow_test_labels = [] for data, target in test_loader: features = model(data) shallow_test_features.append(features) shallow_test_labels.append(target) shallow_test_features = torch.cat(shallow_test_features, dim=0).numpy() shallow_test_labels = torch.cat(shallow_test_labels, dim=0).numpy() shallow_test_features = scaler.transform(shallow_test_features) shallow_acc = accuracy_score(shallow_test_labels, shallow_elm.predict(shallow_test_features)) print("Shallow accuracy:", shallow_acc) # 训练深层特征的ELM分类器 train_loader = torch.utils.data.DataLoader(dataset=train_dataset, batch_size=batch_size, shuffle=True) test_loader = torch.utils.data.DataLoader(dataset=test_dataset, batch_size=batch_size, shuffle=False) model = CNN() criterion = nn.CrossEntropyLoss() optimizer = optim.Adam(model.parameters(), lr=lr) for epoch in range(num_epochs): train(model, train_loader, optimizer, criterion) deep_train_features = [] deep_train_labels = [] for data, target in train_loader: features = model(data) deep_train_features.append(features) deep_train_labels.append(target) deep_train_features = torch.cat(deep_train_features, dim=0).numpy() deep_train_labels = torch.cat(deep_train_labels, dim=0).numpy() scaler = StandardScaler() deep_train_features = scaler.fit_transform(deep_train_features) deep_elm.fit(deep_train_features, deep_train_labels) deep_test_features = [] deep_test_labels = [] for data, target in test_loader: features = model(data) deep_test_features.append(features) deep_test_labels.append(target) deep_test_features = torch.cat(deep_test_features, dim=0).numpy() deep_test_labels = torch.cat(deep_test_labels, dim=0).numpy() deep_test_features = scaler.transform(deep_test_features) deep_acc = accuracy_score(deep_test_labels, deep_elm.predict(deep_test_features)) print("Deep accuracy:", deep_acc) # 训练深浅两个特征的ELM分类器 train_loader = torch.utils.data.DataLoader(dataset=train_dataset, batch_size=batch_size, shuffle=True) test_loader = torch.utils.data.DataLoader(dataset=test_dataset, batch_size=batch_size, shuffle=False) model = CNN() criterion = nn.CrossEntropyLoss() optimizer = optim.Adam(model.parameters(), lr=lr) for epoch in range(num_epochs): train(model, train_loader, optimizer, criterion) shallow_train_features = [] deep_train_features = [] mixed_train_labels = [] for data, target in train_loader: features = model(data) shallow_train_features.append(features[:, :64]) deep_train_features.append(features[:, 64:]) mixed_train_labels.append(target) shallow_train_features = torch.cat(shallow_train_features, dim=0).numpy() deep_train_features = torch.cat(deep_train_features, dim=0).numpy() mixed_train_labels = torch.cat(mixed_train_labels, dim=0).numpy() scaler = StandardScaler() shallow_train_features = scaler.fit_transform(shallow_train_features) deep_train_features = scaler.fit_transform(deep_train_features) mixed_train_features = np.concatenate([shallow_train_features, deep_train_features], axis=1) mixed_elm.fit(mixed_train_features, mixed_train_labels) shallow_test_features = [] deep_test_features = [] mixed_test_labels = [] for data, target in test_loader: features = model(data) shallow_test_features.append(features[:, :64]) deep_test_features.append(features[:, 64:]) mixed_test_labels.append(target) shallow_test_features = torch.cat(shallow_test_features, dim=0).numpy() deep_test_features = torch.cat(deep_test_features, dim=0).numpy() mixed_test_labels = torch.cat(mixed_test_labels, dim=0).numpy() shallow_test_features = scaler.transform(shallow_test_features) deep_test_features = scaler.transform(deep_test_features) mixed_test_features = np.concatenate([shallow_test_features, deep_test_features], axis=1) mixed_acc = accuracy_score(mixed_test_labels, mixed_elm.predict(mixed_test_features)) print("Mixed accuracy:", mixed_acc) ``` 使用上述代码,浅层特征ELM分类器的准确率约为96.5%,深层特征ELM分类器的准确率约为98.2%,深浅两个特征ELM分类器的准确率约为98.5%。训练时间因机器性能而异,一般情况下,深度特征提取的时间较长,ELM分类器的时间较短。
阅读全文

相关推荐

最新推荐

recommend-type

用Pytorch训练CNN(数据集MNIST,使用GPU的方法)

在本文中,我们将探讨如何使用PyTorch训练一个卷积神经网络(CNN)模型,针对MNIST数据集,并利用GPU加速计算。MNIST是一个包含手写数字图像的数据集,常用于入门级的深度学习项目。PyTorch是一个灵活且用户友好的...
recommend-type

Pytorch使用MNIST数据集实现CGAN和生成指定的数字方式

在本教程中,我们将探讨如何使用PyTorch框架来实现条件生成对抗网络(CGAN)并利用MNIST数据集生成指定数字的图像。CGAN是一种扩展了基础生成对抗网络(GAN)的概念,它允许在生成过程中加入额外的条件信息,如类...
recommend-type

Pytorch 使用CNN图像分类的实现

在PyTorch中实现CNN(卷积神经网络)进行图像分类是深度学习中常见的任务,尤其是在计算机视觉领域。本示例中的任务是基于4x4像素的二值图像,目标是根据外围黑色像素点和内圈黑色像素点的数量差异进行分类。如果...
recommend-type

Pytorch提取模型特征向量保存至csv的例子

在PyTorch中,提取模型特征向量并将其保存到CSV文件是一项常见的任务,尤其是在进行图像分类、物体检测或图像分析等应用时。本例子主要展示了如何利用预训练的模型,如ResNet,来提取图像的特征,并将这些特征向量...
recommend-type

在Pytorch中使用Mask R-CNN进行实例分割操作

【PyTorch中使用Mask R-CNN进行实例分割】 实例分割是计算机视觉领域的一个关键任务,它旨在识别图像中每个像素所属的对象实例。不同于语义分割,实例分割不仅标识像素的类别,还能区分同一类的不同实例。Mask R-...
recommend-type

掌握Android RecyclerView拖拽与滑动删除功能

知识点: 1. Android RecyclerView使用说明: RecyclerView是Android开发中经常使用到的一个视图组件,其主要作用是高效地展示大量数据,具有高度的灵活性和可配置性。与早期的ListView相比,RecyclerView支持更加复杂的界面布局,并且能够优化内存消耗和滚动性能。开发者可以对RecyclerView进行自定义配置,如添加头部和尾部视图,设置网格布局等。 2. RecyclerView的拖拽功能实现: RecyclerView通过集成ItemTouchHelper类来实现拖拽功能。ItemTouchHelper类是RecyclerView的辅助类,用于给RecyclerView添加拖拽和滑动交互的功能。开发者需要创建一个ItemTouchHelper的实例,并传入一个实现了ItemTouchHelper.Callback接口的类。在这个回调类中,可以定义拖拽滑动的方向、触发的时机、动作的动画以及事件的处理逻辑。 3. 编辑模式的设置: 编辑模式(也称为拖拽模式)的设置通常用于允许用户通过拖拽来重新排序列表中的项目。在RecyclerView中,可以通过设置Adapter的isItemViewSwipeEnabled和isLongPressDragEnabled方法来分别启用滑动和拖拽功能。在编辑模式下,用户可以长按或触摸列表项来实现拖拽,从而对列表进行重新排序。 4. 左右滑动删除的实现: RecyclerView的左右滑动删除功能同样利用ItemTouchHelper类来实现。通过定义Callback中的getMovementFlags方法,可以设置滑动方向,例如,设置左滑或右滑来触发删除操作。在onSwiped方法中编写处理删除的逻辑,比如从数据源中移除相应数据,并通知Adapter更新界面。 5. 移动动画的实现: 在拖拽或滑动操作完成后,往往需要为项目移动提供动画效果,以增强用户体验。在RecyclerView中,可以通过Adapter在数据变更前后调用notifyItemMoved方法来完成位置交换的动画。同样地,添加或删除数据项时,可以调用notifyItemInserted或notifyItemRemoved等方法,并通过自定义动画资源文件来实现丰富的动画效果。 6. 使用ItemTouchHelperDemo-master项目学习: ItemTouchHelperDemo-master是一个实践项目,用来演示如何实现RecyclerView的拖拽和滑动功能。开发者可以通过这个项目源代码来了解和学习如何在实际项目中应用上述知识点,掌握拖拽排序、滑动删除和动画效果的实现。通过观察项目文件和理解代码逻辑,可以更深刻地领会RecyclerView及其辅助类ItemTouchHelper的使用技巧。
recommend-type

【IBM HttpServer入门全攻略】:一步到位的安装与基础配置教程

# 摘要 本文详细介绍了IBM HttpServer的全面部署与管理过程,从系统需求分析和安装步骤开始,到基础配置与性能优化,再到安全策略与故障诊断,最后通过案例分析展示高级应用。文章旨在为系统管理员提供一套系统化的指南,以便快速掌握IBM HttpServer的安装、配置及维护技术。通过本文的学习,读者能有效地创建和管理站点,确保
recommend-type

[root@localhost~]#mount-tcifs-0username=administrator,password=hrb.123456//192.168.100.1/ygptData/home/win mount:/home/win:挂载点不存在

### CIFS挂载时提示挂载点不存在的解决方案 当尝试通过 `mount` 命令挂载CIFS共享目录时,如果遇到错误提示“挂载点不存在”,通常是因为目标路径尚未创建或者权限不足。以下是针对该问题的具体分析和解决方法: #### 创建挂载点 在执行挂载操作之前,需确认挂载的目标路径已经存在并具有适当的权限。可以使用以下命令来创建挂载点: ```bash mkdir -p /mnt/win_share ``` 上述命令会递归地创建 `/mnt/win_share` 路径[^1]。 #### 配置用户名和密码参数 为了成功连接到远程Windows共享资源,在 `-o` 参数中指定 `user
recommend-type

惠普8594E与IT8500系列电子负载使用教程

在详细解释给定文件中所涉及的知识点之前,需要先明确文档的主题内容。文档标题中提到了两个主要的仪器:惠普8594E频谱分析仪和IT8500系列电子负载。首先,我们将分别介绍这两个设备以及它们的主要用途和操作方式。 惠普8594E频谱分析仪是一款专业级的电子测试设备,通常被用于无线通信、射频工程和微波工程等领域。频谱分析仪能够对信号的频率和振幅进行精确的测量,使得工程师能够观察、分析和测量复杂信号的频谱内容。 频谱分析仪的功能主要包括: 1. 测量信号的频率特性,包括中心频率、带宽和频率稳定度。 2. 分析信号的谐波、杂散、调制特性和噪声特性。 3. 提供信号的时间域和频率域的转换分析。 4. 频率计数器功能,用于精确测量信号频率。 5. 进行邻信道功率比(ACPR)和发射功率的测量。 6. 提供多种输入和输出端口,以适应不同的测试需求。 频谱分析仪的操作通常需要用户具备一定的电子工程知识,对信号的基本概念和频谱分析的技术要求有所了解。 接下来是可编程电子负载,以IT8500系列为例。电子负载是用于测试和评估电源性能的设备,它模拟实际负载的电气特性来测试电源输出的电压和电流。电子负载可以设置为恒流、恒压、恒阻或恒功率工作模式,以测试不同条件下的电源表现。 电子负载的主要功能包括: 1. 模拟各种类型的负载,如电阻性、电感性及电容性负载。 2. 实现负载的动态变化,模拟电流的变化情况。 3. 进行短路测试,检查电源设备在过载条件下的保护功能。 4. 通过控制软件进行远程控制和自动测试。 5. 提供精确的电流和电压测量功能。 6. 通过GPIB、USB或LAN等接口与其他设备进行通信和数据交换。 使用电子负载时,工程师需要了解其操作程序、设置和编程方法,以及如何根据测试目的配置负载参数。 文档的描述部分提到了这些资料的专业性和下载人群的稀少。这可能暗示了这些设备的目标用户是具备一定专业知识的工程师和技术人员,因此文档内容将涵盖较为复杂的操作指南和技术细节。 标签中提到了“中文说明书”,表明这些文件是为中文用户提供方便而制作的,这对于不熟悉英语的技术人员来说是非常重要的。这有助于减少语言障碍,使得中文使用者能够更容易掌握这些专业的测试设备使用方法。 综上所述,惠普8594E频谱分析仪和IT8500系列电子负载都是测试设备领域中不可或缺的工具。掌握它们的使用方法和功能对于电子工程师来说是必需的。这些设备在维护和开发电子系统、电源设备以及无线通信设备中起着至关重要的作用。这份文档对于涉及相关领域的工作技术人员,特别是在中国环境下,提供了非常实用和必需的专业知识。
recommend-type

MATLAB与Python在SAR点目标仿真中的对决:哪种工具更胜一筹?

# 摘要 本文系统地探讨了SAR点目标仿真技术,重点分析了MATLAB和Python在这一领域的应用及其潜在价值。首先概述了SAR仿真概念和需求,然后详细介绍了MATLAB在SAR仿真中的功能、信号处理应用以及仿真实例。接着,文章转向Python,分析了其语言特点、生态系统