# DCGAN

**Repository Path**: lingcb/dcgan

## Basic Information

- **Project Name**: DCGAN
- **Description**: 一个经典的CNN GAN图片生成项目，深度学习框架使用了paddlepaddle2.0。
- **Primary Language**: Python
- **License**: Apache-2.0
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 1
- **Forks**: 0
- **Created**: 2021-12-01
- **Last Updated**: 2024-04-25

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# 前言

最近研究GAN,自然接触到了最经典的DCGAN,想用paddle复现下，在网上搜了下，已经有人复现过了，我也就没再重新造轮子了，直接在别人的项目上做了些修改，然后自己搜集了些数据，训练了一个生成动漫头像的DCGAN.

效果如图：
![请添加图片描述](https://img-blog.csdnimg.cn/c68d739617f5401ea956cbcab9bce0d8.png?x-oss-process=image/watermark,type_d3F5LXplbmhlaQ,shadow_50,text_Q1NETiBA54u45LiN5Yeh,size_20,color_FFFFFF,t_70,g_se,x_16)

# DCGAN开源项目介绍
开源链接：https://gitee.com/lingcb/dcgan

```bash
dcgan
├── LICENSE
├── README.md
├── generator.params
├── main.py
├── output
│   └── imgs
│       ├── 13.jpg
        ...
└── src
    ├── __init__.py
    ├── __pycache__
    │   ├── __init__.cpython-37.pyc
    │   ├── dataset.cpython-37.pyc
    │   ├── model.cpython-37.pyc
    │   └── parameter.cpython-37.pyc
    ├── cutImage.py
    ├── dataset.py
    ├── losses.json
    ├── model.py
    ├── parameter.py
    ├── train.py
    └── work
        └── generator.params
```

# src文件夹
**work文件夹**里的**generator.params 是训练好的模型参数，可以直接加载使用**
**train.py** 是训练代码
**model.py** 里面是生成器和判别器的定义
**cutImage.py** 里面是对原始图片的处理celebA的头像数据处理，我们是直接生成动漫图片，所以这里用不到。
**dataset.py** 里面写了paddle加载数据的类
**parameter.py** 里面定义全局变量
![请添加图片描述](https://img-blog.csdnimg.cn/3f19d1d1a01648dc91c7492593969075.png?x-oss-process=image/watermark,type_d3F5LXplbmhlaQ,shadow_50,text_Q1NETiBA54u45LiN5Yeh,size_20,color_FFFFFF,t_70,g_se,x_16)
# 训练自己的数据
数据可以取百度的aistudio里边找
https://aistudio.baidu.com/aistudio/datasetdetail/17962

要使用自己的代码来训练自己的数据，
自己可以准备好自己的数据放到一个文件夹里，再把**parameters.py里的outpath修改成该文件夹的路径**
然后就可以开始训练了
```bash
cd dcgan/src
python train.py
```

# 使用训练好的模型

```bash
cd dcgan/src
python main.py

#######
运行前，确保python环境装了paddle,cv2
python -m pip install paddlepaddle==2.2.0 -i https://mirror.baidu.com/pypi/simple
pip install opencv-python

程序运行时，会弹出一个显示图片的窗口，
可以使用快捷键进行控制
按 S 会保存当前图片
按 Q 会直接退出程序
按 其他键 会生成下一张
#######
```
**下面是效果**
![请添加图片描述](https://img-blog.csdnimg.cn/11d61219186948bab224f1c4e27a15fb.png?x-oss-process=image/watermark,type_d3F5LXplbmhlaQ,shadow_50,text_Q1NETiBA54u45LiN5Yeh,size_20,color_FFFFFF,t_70,g_se,x_16)
# 参考




[【1】Paddle2.0-通过DCGAN实现人脸图像](https://blog.csdn.net/weixin_42363722/article/details/110870886)