【文件属性】：
文件名称：dcgan-cuda:使用Cuda实现适用于CPU和GPU的深度卷积生成对抗网络
文件大小：34.15MB
文件格式：ZIP
更新时间：2021-05-02 16:40:02
C++
用C ++ / Cuda实现DCGAN的CPU / GPU实现
使用celebA数据集训练的DCGAN的实现，以生成人脸。
要求
对于CUDA版本，您需要：
GCC 8.0或更高版本
CUDA 9.2
至少2.4GB内存（GPU）
建造
您需要下载celebA数据集。 （下一节）
./bootstrap
cd _build
make
celebA数据集
您需要下载celebA数据集并将其放在./celebA中。 然后运行：
cd _build
python ../dcgan/preprocess.py
此命令缩放所有数据集，可能需要一些时间。
直流电源
cd _build
RT_MODE= < mode> ./dcgan
选项：
--model：如果文件存在则加载预训练的模型，并在训练过程中保存模型
--train：训练网络几个时期
--generate：生成一张面Kong