# CVAE

**Repository Path**: olivegame/cvae

## Basic Information

- **Project Name**: CVAE
- **Description**: CVAE条件变分自编码器pytorch代码和原理
- **Primary Language**: Python
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2025-10-09
- **Last Updated**: 2025-10-09

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# CVAE

#### 介绍
CVAE条件变分自编码器pytorch代码和原理

#### 训练结果

- VAE训练结果

![vae](optimized_training_curves_vae.png)

- CVAE训练结果

![cvae](optimized_training_curves_cvae.png)

#### 预测结果

- t-SNE3D可视化结果：潜变量 z 维度为20，加上数字标签10维，进行降维可视化得到0~9数字结果

![t-SNE](t-SNE3D.png)

- 分类效果混合的原因：

![bad](t-sne%E5%88%86%E7%B1%BB%E6%95%88%E6%9E%9C%E4%B8%8D%E5%A5%BD%E7%9A%84%E5%8E%9F%E5%9B%A0.png)

- CVAE预测生成数字gray图片：

![gray](predict/generated_CVAE_100_gray.png)

- CVAE预测生成数字图片：

![predict](predict/generated_CVAE_100.png)

#### 使用说明

1. 代码参考自：https://github.com/timbmg/VAE-CVAE-MNIST

2. predict.py 文件用于生成图片

3. test.py 文件用于生成 t-SNE3D 图片

4. train.py 文件中关键参数 "--conditional" 默认为 CVAE；"--is_tsne" 默认在训练时每个 epoch 生成 t-SNE2D 结果

5. 训练过程所有图片都保存在 figs 文件夹中