DCGAN增强图片数据集

DCGAN增强图片数据集

1.Dependencies

• Python 3.6+

• PyTorch 0.4.0

• numpy 1.14.1, matplotlib 2.2.2, scipy 1.1.0

• imageio 2.3.0

• tqdm 4.24.0

2.DCGAN

步骤：

• 将图片数集放在/Anime_GAN/DCGAN/faces

• 进行如下的命令：

  $ cd Anime_GAN/DCGAN/
  $ python main.py --help # 查看默认参数信息，根据需求可进行修改

执行完上述命令会产生相应的一张图片（存储位 置：/Anime_GAN/DCGAN/saved/img/xx.png）

• 调用SegmentePictures.py进行图片的切割

$ cd DCGAN/saved
$ python SegmentePictures.py　  

# encoding:utf-8
from PIL import Image
import sys
import math
import argparse

def fill_image(image):
    """
    将图片填充为正方形
    :param image:
    :return: