# ColorGAN

**Repository Path**: codeinLinXu/color-gan

## Basic Information

- **Project Name**: ColorGAN
- **Description**: cycleGAN用来做图片上色
- **Primary Language**: Python
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 2
- **Created**: 2022-02-18
- **Last Updated**: 2022-06-17

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

@[TOC](paddle复现colorGAN动漫黑白线条上色)
# 前言
CGAN是生成对抗网络的一个重要的分支，至于CGAN是啥，
可以参考我的另一篇文章[生成对抗网络（GAN）原理详记](https://blog.csdn.net/weixin_45259896/article/details/121733861),里边有CGAN的集体解释。本文主要建我开源的paddle实现的colorGAN项目的使用。

# 项目地址&运行结果
[项目地址](https://gitee.com/lingcb/color-gan)
https://gitee.com/lingcb/color-gan

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

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

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



# ColorGAN 文件目录

```bash
color-gan
├── README.md
├── checkpoints //已经训练好的参数
│   ├── anime512x512_generator.params
│   ├── anime_256x256_color_generator.params
│   └── generator.params
├── dataset  //原始数据
│   ├── blur
│   │   ├── 00deabfe1f0d7c8b5ef4ca1d8c314925.jpg
...
│   │   └── 0a4a93968e913bd9f7d7bce7f92494f0.jpg
│   ├── edge
│   │   ├── 00deabfe1f0d7c8b5ef4ca1d8c314925.jpg
...
│   │   └── 0a3e05b6584deece72c0c0000ee07fcf.jpg
│   └── origin
│       ├── 00deabfe1f0d7c8b5ef4ca1d8c314925.jpg
...
│       └── 0a4a93968e913bd9f7d7bce7f92494f0.jpg
├── main.py
├── output
│   └── imgs  //上色的一些结果
│       ├── 0.jpg
...
│       └── 7.jpg
└── src
    ├── __init__.py
    ├── __pycache__
    │   ├── __init__.cpython-37.pyc
    │   ├── dataset.cpython-37.pyc
    │   ├── model.cpython-37.pyc
    │   └── parameter.cpython-37.pyc
    ├── cutImage.py  //将原始图片转换成edge,blur，然后resize
    ├── dataset.py   //paddle dataset 数据架子啊类
    ├── losses.json  
    ├── mix.py
    ├── model.py      //生成器与判别的网络模型代码
    ├── parameter.py  //一些全局初始参数
    ├── train.py      //训练代码
    └── work
        └── Output
```

# 测试
```bash
#安装一些相关的第三方包
python -m pip install paddlepaddle==2.2.0 -i https://mirror.baidu.com/pypi/simple
pip install opencv-python

#克隆源码并运行
git clone https://gitee.com/lingcb/color-gan.git
cd color-gan
python main.py
```

# 训练自己的数据集
自己准备好自己的数据集集
参照下面结构放置,可以使用**cutImage.py**里的代码辅助制作自己的数据集
```bash
├── dataset  //原始数据
│   ├── blur
│   │   ├── 00deabfe1f0d7c8b5ef4ca1d8c314925.jpg
...
│   │   └── 0a4a93968e913bd9f7d7bce7f92494f0.jpg
│   ├── edge
│   │   ├── 00deabfe1f0d7c8b5ef4ca1d8c314925.jpg
...
│   │   └── 0a3e05b6584deece72c0c0000ee07fcf.jpg
│   └── origin
│       ├── 00deabfe1f0d7c8b5ef4ca1d8c314925.jpg
...
│       └── 0a4a93968e913bd9f7d7bce7f92494f0.jpg
```

然后，在修改下，parameter.py里的outpath,就可一开始训练了
```bash
#安装一些相关的第三方包
python -m pip install paddlepaddle==2.2.0 -i https://mirror.baidu.com/pypi/simple
pip install opencv-python

#克隆源码并运行
git clone https://gitee.com/lingcb/color-gan.git
cd color-gan/src
python train.py
```

# 参考
【1】[Paddle2.0-通过DCGAN实现人脸图像](https://blog.csdn.net/weixin_42363722/article/details/110870886)
【2】[【pytorch基础笔记五】基于条件GAN的色彩填充](https://blog.csdn.net/Fredric_2014/article/details/107567835)