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

时间: 2024-05-05 07:23:00 浏览: 118
下面是使用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

Spring Websocket快速实现与SSMTest实战应用

标题“websocket包”指代的是一个在计算机网络技术中应用广泛的组件或技术包。WebSocket是一种网络通信协议,它提供了浏览器与服务器之间进行全双工通信的能力。具体而言,WebSocket允许服务器主动向客户端推送信息,是实现即时通讯功能的绝佳选择。 描述中提到的“springwebsocket实现代码”,表明该包中的核心内容是基于Spring框架对WebSocket协议的实现。Spring是Java平台上一个非常流行的开源应用框架,提供了全面的编程和配置模型。在Spring中实现WebSocket功能,开发者通常会使用Spring提供的注解和配置类,简化WebSocket服务端的编程工作。使用Spring的WebSocket实现意味着开发者可以利用Spring提供的依赖注入、声明式事务管理、安全性控制等高级功能。此外,Spring WebSocket还支持与Spring MVC的集成,使得在Web应用中使用WebSocket变得更加灵活和方便。 直接在Eclipse上面引用,说明这个websocket包是易于集成的库或模块。Eclipse是一个流行的集成开发环境(IDE),支持Java、C++、PHP等多种编程语言和多种框架的开发。在Eclipse中引用一个库或模块通常意味着需要将相关的jar包、源代码或者配置文件添加到项目中,然后就可以在Eclipse项目中使用该技术了。具体操作可能包括在项目中添加依赖、配置web.xml文件、使用注解标注等方式。 标签为“websocket”,这表明这个文件或项目与WebSocket技术直接相关。标签是用于分类和快速检索的关键字,在给定的文件信息中,“websocket”是核心关键词,它表明该项目或文件的主要功能是与WebSocket通信协议相关的。 文件名称列表中的“SSMTest-master”暗示着这是一个版本控制仓库的名称,例如在GitHub等代码托管平台上。SSM是Spring、SpringMVC和MyBatis三个框架的缩写,它们通常一起使用以构建企业级的Java Web应用。这三个框架分别负责不同的功能:Spring提供核心功能;SpringMVC是一个基于Java的实现了MVC设计模式的请求驱动类型的轻量级Web框架;MyBatis是一个支持定制化SQL、存储过程以及高级映射的持久层框架。Master在这里表示这是项目的主分支。这表明websocket包可能是一个SSM项目中的模块,用于提供WebSocket通讯支持,允许开发者在一个集成了SSM框架的Java Web应用中使用WebSocket技术。 综上所述,这个websocket包可以提供给开发者一种简洁有效的方式,在遵循Spring框架原则的同时,实现WebSocket通信功能。开发者可以利用此包在Eclipse等IDE中快速开发出支持实时通信的Web应用,极大地提升开发效率和应用性能。
recommend-type

电力电子技术的智能化:数据中心的智能电源管理

# 摘要 本文探讨了智能电源管理在数据中心的重要性,从电力电子技术基础到智能化电源管理系统的实施,再到技术的实践案例分析和未来展望。首先,文章介绍了电力电子技术及数据中心供电架构,并分析了其在能效提升中的应用。随后,深入讨论了智能化电源管理系统的组成、功能、监控技术以及能
recommend-type

通过spark sql读取关系型数据库mysql中的数据

Spark SQL是Apache Spark的一个模块,它允许用户在Scala、Python或SQL上下文中查询结构化数据。如果你想从MySQL关系型数据库中读取数据并处理,你可以按照以下步骤操作: 1. 首先,你需要安装`PyMySQL`库(如果使用的是Python),它是Python与MySQL交互的一个Python驱动程序。在命令行输入 `pip install PyMySQL` 来安装。 2. 在Spark环境中,导入`pyspark.sql`库,并创建一个`SparkSession`,这是Spark SQL的入口点。 ```python from pyspark.sql imp
recommend-type

新版微软inspect工具下载:32位与64位版本

根据给定文件信息,我们可以生成以下知识点: 首先,从标题和描述中,我们可以了解到新版微软inspect.exe与inspect32.exe是两个工具,它们分别对应32位和64位的系统架构。这些工具是微软官方提供的,可以用来下载获取。它们源自Windows 8的开发者工具箱,这是一个集合了多种工具以帮助开发者进行应用程序开发与调试的资源包。由于这两个工具被归类到开发者工具箱,我们可以推断,inspect.exe与inspect32.exe是用于应用程序性能检测、问题诊断和用户界面分析的工具。它们对于开发者而言非常实用,可以在开发和测试阶段对程序进行深入的分析。 接下来,从标签“inspect inspect32 spy++”中,我们可以得知inspect.exe与inspect32.exe很有可能是微软Spy++工具的更新版或者是有类似功能的工具。Spy++是Visual Studio集成开发环境(IDE)的一个组件,专门用于Windows应用程序。它允许开发者观察并调试与Windows图形用户界面(GUI)相关的各种细节,包括窗口、控件以及它们之间的消息传递。使用Spy++,开发者可以查看窗口的句柄和类信息、消息流以及子窗口结构。新版inspect工具可能继承了Spy++的所有功能,并可能增加了新功能或改进,以适应新的开发需求和技术。 最后,由于文件名称列表仅提供了“ed5fa992d2624d94ac0eb42ee46db327”,没有提供具体的文件名或扩展名,我们无法从这个文件名直接推断出具体的文件内容或功能。这串看似随机的字符可能代表了文件的哈希值或是文件存储路径的一部分,但这需要更多的上下文信息来确定。 综上所述,新版的inspect.exe与inspect32.exe是微软提供的开发者工具,与Spy++有类似功能,可以用于程序界面分析、问题诊断等。它们是专门为32位和64位系统架构设计的,方便开发者在开发过程中对应用程序进行深入的调试和优化。同时,使用这些工具可以提高开发效率,确保软件质量。由于这些工具来自Windows 8的开发者工具箱,它们可能在兼容性、效率和用户体验上都经过了优化,能够为Windows应用的开发和调试提供更加专业和便捷的解决方案。
recommend-type

如何运用电力电子技术实现IT设备的能耗监控

# 摘要 随着信息技术的快速发展,IT设备能耗监控已成为提升能效和减少环境影响的关键环节。本文首先概述了电力电子技术与IT设备能耗监控的重要性,随后深入探讨了电力电子技术的基础原理及其在能耗监控中的应用。文章详细分析了IT设备能耗监控的理论框架、实践操作以及创新技术的应用,并通过节能改造案例展示了监控系统构建和实施的成效。最后,本文展望了未来能耗监控技术的发展趋势,同时