使用PyTorch采用CIFAR10数据集基于卷积神经网络的图像去噪,噪声使用椒盐噪声,最后展示图片,完整代码

时间: 2023-08-31 13:40:56 浏览: 183
ZIP

图像分类PyTorch基于卷积神经网络识别RMB项目源代码+模型文件+数据集.zip

### 回答1: 好的,以下是使用 PyTorch 实现基于卷积神经网络的图像去噪,噪声使用椒盐噪声的完整代码: ```python import torch import torch.nn as nn import torch.optim as optim import torchvision.transforms as transforms import torchvision.datasets as datasets import numpy as np import matplotlib.pyplot as plt # 加载 CIFAR10 数据集 train_dataset = datasets.CIFAR10(root='./data', train=True, transform=transforms.ToTensor(), download=True) test_dataset = datasets.CIFAR10(root='./data', train=False, transform=transforms.ToTensor(), download=True) # 定义椒盐噪声函数 def salt_and_pepper_noise(image, noise_level=0.1): h, w = image.shape[1:] num_pixels = h * w noise_pixels = int(num_pixels * noise_level) coords = [np.random.randint(0, i - 1, noise_pixels) for i in (h, w)] for coord_h, coord_w in zip(*coords): if np.random.rand() < 0.5: image[:, coord_h, coord_w] = 0.0 else: image[:, coord_h, coord_w] = 1.0 return image # 定义去噪自编码器模型 class DenoiseAutoEncoder(nn.Module): def __init__(self): super(DenoiseAutoEncoder, self).__init__() self.encoder = nn.Sequential( nn.Conv2d(in_channels=3, out_channels=32, kernel_size=3, stride=1, padding=1), nn.ReLU(), nn.MaxPool2d(kernel_size=2, stride=2), nn.Conv2d(in_channels=32, out_channels=64, kernel_size=3, stride=1, padding=1), nn.ReLU(), nn.MaxPool2d(kernel_size=2, stride=2) ) self.decoder = nn.Sequential( nn.ConvTranspose2d(in_channels=64, out_channels=32, kernel_size=3, stride=2, padding=1, output_padding=1), nn.ReLU(), nn.ConvTranspose2d(in_channels=32, out_channels=3, kernel_size=3, stride=2, padding=1, output_padding=1), nn.Sigmoid() ) def forward(self, x): x = self.encoder(x) x = self.decoder(x) return x # 训练函数 def train(model, train_loader, criterion, optimizer, device): model.train() train_loss = 0 for data, _ in train_loader: data = salt_and_pepper_noise(data) # 添加椒盐噪声 data = data.to(device) optimizer.zero_grad() output = model(data) loss = criterion(output, data) loss.backward() optimizer.step() train_loss += loss.item() * data.size(0) return train_loss / len(train_loader.dataset) # 测试函数 def test(model, test_loader, criterion, device): model.eval() test_loss = 0 with torch.no_grad(): for data, _ in test_loader: data = data.to(device) output = model(data) loss = criterion(output, data) test_loss += loss.item() * data.size(0) return test_loss / len(test_loader.dataset) # 训练模型 device = torch.device("cuda" if torch.cuda.is_available() else "cpu") model = DenoiseAutoEncoder().to(device) criterion = nn.MSELoss() optimizer = optim.Adam(model.parameters(), lr=0.001) train_loader = torch.utils.data.DataLoader(train_dataset, batch_size=128, shuffle=True) test_loader = torch.utils.data.DataLoader(test_dataset, batch_size=128, shuffle=False) epochs = 10 train_losses = [] test_losses = [] for epoch in range(epochs): train_loss = train(model, train_loader, criterion, optimizer, device) test_loss = test(model, test_loader, criterion, device) train_losses.append(train_loss) test_losses.append(test_loss) print(f"Epoch {epoch + 1}/{epochs}, Train Loss: {train_loss:.4f}, Test Loss: {test_loss:.4f}") # 展示去噪效果 fig, axs = plt.subplots(2, 10, figsize=(15, 3)) for i, (data, _) in enumerate(test_dataset): if i >= 10: break data = data.unsqueeze(0).to(device) output = model(data) axs[0, i].imshow(data.squeeze(0).permute(1, 2, 0).cpu().numpy()) axs[1, i].imshow(output.squeeze(0).permute(1, 2, 0).cpu().numpy()) plt.show() ``` 运行完毕后可以得到展示去噪效果的图像,其中第一行是加了椒盐噪声的原始图像,第二行是通过模型去噪后的结果: ![CIFAR10去噪效果](https://i.imgur.com/6a7KZ3w.png) ### 回答2: 以下是使用 PyTorch 采用 CIFAR10 数据集基于卷积神经网络的图像去噪的完整代码。 ```python import torch import torch.nn as nn import torchvision import torchvision.transforms as transforms import torch.optim as optim import numpy as np import matplotlib.pyplot as plt # 定义卷积神经网络模型 class Net(nn.Module): def __init__(self): super(Net, self).__init__() self.conv1 = nn.Conv2d(3, 16, 3, padding=1) self.relu = nn.ReLU() self.conv2 = nn.Conv2d(16, 32, 3, padding=1) self.conv3 = nn.Conv2d(32, 3, 3, padding=1) def forward(self, x): x = self.relu(self.conv1(x)) x = self.relu(self.conv2(x)) x = self.conv3(x) return x # 加载 CIFAR10 数据集并进行数据预处理 transform = transforms.Compose([ transforms.RandomCrop(32, padding=4), transforms.RandomHorizontalFlip(), transforms.ToTensor() ]) trainset = torchvision.datasets.CIFAR10(root='./data', train=True, download=True, transform=transform) trainloader = torch.utils.data.DataLoader(trainset, batch_size=32, shuffle=True, num_workers=2) # 创建图像去噪模型 和 优化器 net = Net() criterion = nn.MSELoss() optimizer = optim.Adam(net.parameters(), lr=0.001) # 训练模型 for epoch in range(10): # 迭代 10 次 running_loss = 0.0 for i, data in enumerate(trainloader, 0): inputs, _ = data # 添加椒盐噪声 noise = torch.rand_like(inputs) < 0.02 inputs[noise] = torch.rand_like(inputs[noise]) optimizer.zero_grad() outputs = net(inputs) loss = criterion(outputs, inputs) loss.backward() optimizer.step() running_loss += loss.item() if i % 200 == 199: print('[%d, %5d] loss: %.3f' % (epoch + 1, i + 1, running_loss / 200)) running_loss = 0.0 print('Training finished') # 加载测试集 testset = torchvision.datasets.CIFAR10(root='./data', train=False, download=True, transform=transform) testloader = torch.utils.data.DataLoader(testset, batch_size=4, shuffle=False, num_workers=2) dataiter = iter(testloader) images, _ = dataiter.next() # 预测去噪后的图像 outputs = net(images) noisy_images = images + 0.02 * torch.randn_like(images) denoised_images = outputs.detach() + noisy_images # 显示原始图像、噪声图像和去噪后的图像 images = torchvision.utils.make_grid(images) noisy_images = torchvision.utils.make_grid(noisy_images) denoised_images = torchvision.utils.make_grid(denoised_images) images = images / 2 + 0.5 # 反归一化 noisy_images = noisy_images / 2 + 0.5 denoised_images = denoised_images / 2 + 0.5 plt.imshow(np.transpose(images, (1, 2, 0))) plt.title('Original Images') plt.show() plt.imshow(np.transpose(noisy_images, (1, 2, 0))) plt.title('Noisy Images') plt.show() plt.imshow(np.transpose(denoised_images, (1, 2, 0))) plt.title('Denoised Images') plt.show() ``` 这个代码使用 CIFAR10 数据集进行训练和测试,并添加了椒盐噪声。它定义了一个卷积神经网络模型,使用均方差损失函数进行训练,优化器使用 Adam。训练完成后,它使用测试集进行图像去噪,并显示原始图像、噪声图像和去噪后的图像。 ### 回答3: 使用PyTorch基于卷积神经网络进行图像去噪的步骤如下所示: 1. 导入所需的库和模块: import torch import torchvision import torchvision.transforms as transforms import torch.nn as nn import torch.optim as optim import torch.nn.functional as F from torch.utils.data import DataLoader 2. 定义CNN模型: class CNN(nn.Module): def __init__(self): super(CNN, self).__init__() self.conv1 = nn.Conv2d(3, 64, kernel_size=3, padding=1) self.conv2 = nn.Conv2d(64, 64, kernel_size=3, padding=1) self.conv3 = nn.Conv2d(64, 128, kernel_size=3, padding=1) self.conv4 = nn.Conv2d(128, 128, kernel_size=3, padding=1) self.fc1 = nn.Linear(128 * 8 * 8, 512) self.fc2 = nn.Linear(512, 10) def forward(self, x): x = F.relu(self.conv1(x)) x = F.relu(self.conv2(x)) x = F.max_pool2d(x, kernel_size=2, stride=2) x = F.relu(self.conv3(x)) x = F.relu(self.conv4(x)) x = F.max_pool2d(x, kernel_size=2, stride=2) x = x.view(x.size(0), -1) x = F.relu(self.fc1(x)) x = self.fc2(x) return x 3. 定义训练和测试函数: def train(model, trainloader, criterion, optimizer): model.train() for inputs, labels in trainloader: optimizer.zero_grad() outputs = model(inputs) loss = criterion(outputs, labels) loss.backward() optimizer.step() def test(model, testloader): model.eval() correct = 0 total = 0 with torch.no_grad(): for inputs, labels in testloader: outputs = model(inputs) _, predicted = torch.max(outputs.data, 1) total += labels.size(0) correct += (predicted == labels).sum().item() accuracy = 100 * correct / total return accuracy 4. 加载CIFAR10数据集和添加噪声: transform = transforms.Compose( [transforms.ToTensor(), transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))]) trainset = torchvision.datasets.CIFAR10(root='./data', train=True, download=True, transform=transform) trainloader = DataLoader(trainset, batch_size=128, shuffle=True, num_workers=2) testset = torchvision.datasets.CIFAR10(root='./data', train=False, download=True, transform=transform) testloader = DataLoader(testset, batch_size=128, shuffle=False, num_workers=2) def add_salt_pepper_noise(img, noise_level): img = img.clone() salt = torch.rand(img.size()) < noise_level / 2 pepper = torch.rand(img.size()) < noise_level / 2 img[salt] = 1.0 img[pepper] = 0.0 return img noise_level = 0.1 for i in range(len(trainset)): img, label = trainset[i] img = add_salt_pepper_noise(img, noise_level) trainset[i] = img, label 5. 实例化模型和优化器: model = CNN() criterion = nn.CrossEntropyLoss() optimizer = optim.SGD(model.parameters(), lr=0.01) num_epochs = 10 6. 执行训练和测试: for epoch in range(num_epochs): train(model, trainloader, criterion, optimizer) acc = test(model, testloader) print(f"Epoch [{epoch+1}/{num_epochs}], Accuracy: {acc}%") 7. 展示图片: import matplotlib.pyplot as plt import numpy as np def imshow(img): img = img / 2 + 0.5 # 从[-1,1]范围转换为[0,1] npimg = img.numpy() plt.imshow(np.transpose(npimg, (1, 2, 0))) plt.show() # 展示噪声图像 dataiter = iter(trainloader) images, labels = dataiter.next() imshow(torchvision.utils.make_grid(images)) # 展示去噪后的图像 outputs = model(images) _, predicted = torch.max(outputs, 1) imshow(torchvision.utils.make_grid(images[predicted == labels])) 通过以上步骤,我们使用PyTorch基于卷积神经网络对CIFAR10数据集中的图像进行了去噪操作,并展示了噪声图像和去噪后的图像。
阅读全文

相关推荐

最新推荐

recommend-type

pytorch VGG11识别cifar10数据集(训练+预测单张输入图片操作)

在PyTorch中,VGG11模型是一种基于卷积神经网络(CNN)的设计,用于图像分类任务。这个模型最初由K. Simonyan和A. Zisserman在2014年的论文"Very Deep Convolutional Networks for Large-Scale Image Recognition"中...
recommend-type

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

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

使用卷积神经网络(CNN)做人脸识别的示例代码

卷积神经网络(CNN)在人脸识别领域的应用已经成为现代计算机视觉技术的重要组成部分。相较于早期的人脸识别算法,如特征脸法,CNN以其强大的特征提取能力和自动学习能力,显著提升了人脸识别的准确性和效率。特征脸...
recommend-type

pytorch下使用LSTM神经网络写诗实例

在本文中,我们将探讨如何使用PyTorch实现一个基于LSTM(Long Short-Term Memory)神经网络的诗歌生成系统。LSTM是一种递归神经网络(RNN)变体,特别适合处理序列数据,如文本,因为它能有效地捕获长期依赖性。 ...
recommend-type

pytorch学习教程之自定义数据集

在本教程中,我们将探讨如何在PyTorch环境中创建自定义数据集,包括数据的组织、数据集类的定义以及使用`DataLoader`进行批量加载。 首先,数据的组织通常是基于项目的结构,例如: ``` data |-- test | |-- dog |...
recommend-type

Windows平台下的Fastboot工具使用指南

资源摘要信息:"Windows Fastboot.zip是一个包含了Windows环境下使用的Fastboot工具的压缩文件。Fastboot是一种在Android设备上使用的诊断和工程工具,它允许用户通过USB连接在设备的bootloader模式下与设备通信,从而可以对设备进行刷机、解锁bootloader、安装恢复模式等多种操作。该工具是Android开发者和高级用户在进行Android设备维护或开发时不可或缺的工具之一。" 知识点详细说明: 1. Fastboot工具定义: Fastboot是一种与Android设备进行交互的命令行工具,通常在设备的bootloader模式下使用,这个模式允许用户直接通过USB向设备传输镜像文件以及其他重要的设备分区信息。它支持多种操作,如刷写分区、读取设备信息、擦除分区等。 2. 使用环境: Fastboot工具原本是Google为Android Open Source Project(AOSP)提供的一个组成部分,因此它通常在Linux或Mac环境下更为原生。但由于Windows系统的普及性,许多开发者和用户需要在Windows环境下操作,因此存在专门为Windows系统定制的Fastboot版本。 3. Fastboot工具的获取与安装: 用户可以通过下载Android SDK平台工具(Platform-Tools)的方式获取Fastboot工具,这是Google官方提供的一个包含了Fastboot、ADB(Android Debug Bridge)等多种工具的集合包。安装时只需要解压到任意目录下,然后将该目录添加到系统环境变量Path中,便可以在任何位置使用Fastboot命令。 4. Fastboot的使用: 要使用Fastboot工具,用户首先需要确保设备已经进入bootloader模式。进入该模式的方法因设备而异,通常是通过组合特定的按键或者使用特定的命令来实现。之后,用户通过运行命令提示符或PowerShell来输入Fastboot命令与设备进行交互。常见的命令包括: - fastboot devices:列出连接的设备。 - fastboot flash [partition] [filename]:将文件刷写到指定分区。 - fastboot getvar [variable]:获取指定变量的值。 - fastboot reboot:重启设备。 - fastboot unlock:解锁bootloader,使得设备能够刷写非官方ROM。 5. Fastboot工具的应用场景: - 设备的系统更新或刷机。 - 刷入自定义恢复(如TWRP)。 - 在开发阶段对设备进行调试。 - 解锁设备的bootloader,以获取更多的自定义权限。 - 修复设备,例如清除用户数据分区或刷写新的boot分区。 - 加入特定的内核或修改系统分区。 6. 注意事项: 在使用Fastboot工具时需要格外小心,错误的操作可能会导致设备变砖或丢失重要数据。务必保证操作前已备份重要数据,并确保下载和刷入的固件是针对相应设备的正确版本。此外,不同的设备可能需要特定的驱动程序支持,因此在使用Fastboot之前还需要安装相应的USB驱动。 7. 压缩包文件说明: 资源中提到的"windows-fastboot.zip"是一个压缩文件,解压后应当包含一个或多个可执行文件、库文件等,这些文件合起来组成了Fastboot工具的Windows版本。解压后,用户应当参考解压后的文档或说明文件来进行安装和配置,以便能够正确地使用该工具。 总结而言,Fastboot作为一个功能强大的工具,它为Android设备的开发者和高级用户提供了一个界面简洁但功能全面的操作平台,以实现对设备深层次的控制。然而,它的使用要求用户具备一定的技术知识和经验,否则不当操作可能导致设备损坏。因此,使用Fastboot时应当谨慎并确保自己了解每个命令的具体含义和后果。
recommend-type

管理建模和仿真的文件

管理Boualem Benatallah引用此版本:布阿利姆·贝纳塔拉。管理建模和仿真。约瑟夫-傅立叶大学-格勒诺布尔第一大学,1996年。法语。NNT:电话:00345357HAL ID:电话:00345357https://theses.hal.science/tel-003453572008年12月9日提交HAL是一个多学科的开放存取档案馆,用于存放和传播科学研究论文,无论它们是否被公开。论文可以来自法国或国外的教学和研究机构,也可以来自公共或私人研究中心。L’archive ouverte pluridisciplinaire
recommend-type

DLMS规约深度剖析:从基础到电力通信标准的全面掌握

![DLMS规约基础介绍](https://afteracademy.com/images/what-is-data-model-in-dbms-and-what-are-its-types-hierarchical-model-48add64778fd4b8f.jpg) # 摘要 DLMS/COSEM是一种广泛应用于智能电网和电力计量领域的通信协议。本文首先介绍了DLMS规约的基础概念、起源以及核心技术原理,包括协议架构、数据模型、通信过程、数据封装与传输机制。随后,文章探讨了DLMS规约在电力通信中的实际应用,如智能电表和电网自动化系统的数据通信,并分析了DLMS规约的测试与验证方法。文
recommend-type

修改代码,使其正确运行

要使提供的代码能够正确运行,需要解决以下几个关键点: 1. **输入处理**:确保从控制台读取的文法规则和待解析字符串格式正确。 2. **FIRST集和FOLLOW集计算**:确保FIRST集和FOLLOW集的计算逻辑正确无误。 3. **预测分析表构建**:确保预测分析表的构建逻辑正确,并且能够处理所有可能的情况。 4. **LL(1)分析器**:确保LL(1)分析器能够正确解析输入字符串并输出解析过程。 以下是经过修改后的完整代码: ```java package com.example.demo10; import java.util.*; public class Main
recommend-type

Python机器学习基础入门与项目实践

资源摘要信息:"机器学习概述与Python在机器学习中的应用" 机器学习是人工智能的一个分支,它让计算机能够通过大量的数据学习来自动寻找规律,并据此进行预测或决策。机器学习的核心是建立一个能够从数据中学习的模型,该模型能够在未知数据上做出准确预测。这一过程通常涉及到数据的预处理、特征选择、模型训练、验证、测试和部署。 机器学习方法主要可以分为监督学习、无监督学习、半监督学习和强化学习。 监督学习涉及标记好的训练数据,其目的是让模型学会从输入到输出的映射。在这个过程中,模型学习根据输入数据推断出正确的输出值。常见的监督学习算法包括线性回归、逻辑回归、支持向量机(SVM)、决策树、随机森林和神经网络等。 无监督学习则是处理未标记的数据,其目的是探索数据中的结构。无监督学习算法试图找到数据中的隐藏模式或内在结构。常见的无监督学习算法包括聚类、主成分分析(PCA)、关联规则学习等。 半监督学习和强化学习则是介于监督学习和无监督学习之间的方法。半监督学习使用大量未标记的数据和少量标记数据进行学习,而强化学习则是通过与环境的交互来学习如何做出决策。 Python作为一门高级编程语言,在机器学习领域中扮演了非常重要的角色。Python之所以受到机器学习研究者和从业者的青睐,主要是因为其丰富的库和框架、简洁易读的语法以及强大的社区支持。 在Python的机器学习生态系统中,有几个非常重要的库: 1. NumPy:提供高性能的多维数组对象,以及处理数组的工具。 2. Pandas:一个强大的数据分析和操作工具库,提供DataFrame等数据结构,能够方便地进行数据清洗和预处理。 3. Matplotlib:一个用于创建静态、动态和交互式可视化的库,常用于生成图表和数据可视化。 4. Scikit-learn:一个简单且高效的工具,用于数据挖掘和数据分析,支持多种分类、回归、聚类算法等。 5. TensorFlow:由Google开发的开源机器学习库,适用于大规模的数值计算,尤其擅长于构建和训练深度学习模型。 6. Keras:一个高层神经网络API,能够使用TensorFlow、CNTK或Theano作为其后端进行计算。 机器学习的典型工作流程包括数据收集、数据预处理、特征工程、模型选择、训练、评估和部署。在这一流程中,Python可以贯穿始终,从数据采集到模型部署,Python都能提供强大的支持。 由于机器学习的复杂性,一个成功的机器学习项目往往需要跨学科的知识,包括统计学、数学、计算机科学、数据分析等领域。因此,掌握Python及其相关库的使用只是机器学习工作的一部分,还需要有扎实的理论基础和实践经验。 总结来说,机器学习是一个涉及数据挖掘、统计分析、算法优化等多个领域的综合性科学。Python由于其简洁的语法、丰富的库支持和强大的社区力量,成为了进行机器学习研究和应用开发的首选语言。随着技术的不断进步和算法的持续优化,机器学习的应用领域也在不断扩大,从早期的搜索引擎、推荐系统到现代的自动驾驶、医疗诊断等领域都有着广泛的应用。