Pytorch 基于Minst手写字的自编码去噪

正确理解图像信息在医学等领域是至关重要的。去噪可以集中在清理旧的扫描图像上，或者有助于癌症生物学中的特征选择。噪音的存在可能会混淆疾病的识别和分析，从而导致不必要的死亡。因此，医学图像去噪是一项必不可少的预处理技术。

所谓的自编码器技术已被证明是非常有用的图像去噪。

自编码器由编码器模型和解码器模型两个相互连接的人工神经网络组成。自动编码器的目标是找到一种将输入图像编码为压缩格式(也称为潜在空间)的方法，使解码后的图像版本尽可能接近输入图像。

Autoencoders如何工作

该网络提供了原始图像x，以及它们的噪声版本x~。该网络试图重构其输出x '，使其尽可能接近原始图像x。通过这样做，它学会了如何去噪图像。

如图所示，编码器模型将输入转换为一个小而密集的表示。解码器模型可以看作是一个生成模型，它能够生成特定的特征。

编码器和解码器网络通常作为一个整体进行训练。损失函数判断网络创建的输出x '与原始输入x的差别。

通过这样做，编码器学会了在有限的潜在空间中保留尽可能多的相关信息，并巧妙地丢弃不相关的部分，如噪声。解码器学习采取压缩潜在信息，并重建它成为一个完全无错误的输入。

导入数据集

import torch
import torch.nn as nn
import torch.optim as optim
from torchvision import datasets
from torchvision import transforms
import torch.nn.functional as F
import torch.utils.data as Data
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd


transform1 = transforms.Compose([
    transforms.ToTensor(),
    # 高斯模糊
    transforms.GaussianBlur(kernel_size=3, sigma=(5.0, 10.0)),  
])

# 获取手写数字的训练集和测试集
train_dataset = datasets.MNIST(root='./zibianma',
                               transform=transform1,
                               train=True,