StyleGAN2预训练微调(实践5)

1.设备

内存
StyleGAN2预训练微调(实践5)
GPU
StyleGAN2预训练微调(实践5)
硬盘:3.5T

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

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

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

3.参数修改
StyleGAN2预训练微调(实践5)
主要修改
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-****博客
StyleGAN2预训练微调(实践5)
查看Metrics
6.下载图片查看效果
StyleGAN2预训练微调(实践5)
7.使用生成生成图片
StyleGAN2预训练微调(实践5)
因为时间过长,所以过拟合了
StyleGAN2预训练微调(实践5)
微信号:herry7788