Skip to content

Latest commit

 

History

History
34 lines (25 loc) · 665 Bytes

README.md

File metadata and controls

34 lines (25 loc) · 665 Bytes

机器学习导引结课论文

机器学习导引结课论文的项目。

项目介绍

学习GAN,将GAN应用于mnist和cub数据集,生成新的图片,用来进行数据增强。

运行

数据集放在根目录的data文件夹下

目录结构

|-- data
    |-- mnist
    |-- CUB_200_2011
        |-- train
|-- data_utils.py
|-- gan.py
|-- my_dataset.py

运行指令

python gan.py

结果截图

gan 用在cub的生成图片

cub生成图片

gan 用在mnist的生成图片

mnist生成图片

参考链接

核心代码gan.py参考自https://github.com/eriklindernoren/PyTorch-GAN