StyleGAN2预训练微调(实践5)

1.设备

内存
在这里插入图片描述
GPU
StyleGAN2预训练微调(实践5)_第1张图片
硬盘:3.5T

2.数据集来自北邮 文铱的新数据集

共13061张图片,resize为256*256后

数据处理参照https://zhuanlan.zhihu.com/p/113156395

3.参数修改
在这里插入图片描述
主要修改
127
128
131 在官方的人脸模型的基础上训练
132 从10000kimg开始训练
135 要不要重新构建一个网络
4.训练命令
python run_training.py --num-gpus=8 --data-dir=/home/data/stylegan/stylegan2 --config=config-f --dataset=dataset --mirror-augment=true
5.查看tensorboard
本地查看远程服务器的tensorboard,参考(MobaXterm强烈推荐下)
本地浏览器远程访问服务器tensorboard(MobaXterm)_运维_Darlewo-CSDN博客
StyleGAN2预训练微调(实践5)_第2张图片
查看Metrics
6.下载图片查看效果
StyleGAN2预训练微调(实践5)_第3张图片
7.使用生成生成图片
StyleGAN2预训练微调(实践5)_第4张图片
因为时间过长,所以过拟合了
StyleGAN2预训练微调(实践5)_第5张图片
微信号:herry7788

你可能感兴趣的:(stylegan2,Stylegan)