VAE变分自编码器

自编码器 

VAE变分自编码器_第1张图片

VAE变分自编码器_第2张图片

变分自编码器

VAE变分自编码器_第3张图片

自编码器是从数据中学习一组向量,但是向量的分布是未知的。

变分自编码器是首先是编码器从数据中学习到一个高斯分布,分别对应上图中均值向量和方差向量。然后解码器在高斯分布随机采样一个向量(利用的重参数化),生成原始图像。

VAE变分自编码器_第4张图片

import torch
import torch.nn as nn
import torch.nn.functional as F
import torchvision
from torchvision import transforms
from torchvision.utils import save_image  # Save a given Tensor into an image file.
from torch.utils.data import DataLoader
from matplotlib import pyplot as plt
import nu

你可能感兴趣的:(深度学习,深度学习,人工智能)