stylegan训练

数据集转换

python dataset_tool.py create_from_images datasets/custom_datasets dataset

datasets/custom_datasets 训练数据保存路径
dataset 图片路径

训练

python train.py --outdir=output --gpus=1 --data=datasets/custom_datasets --snap=25

生成图像

 python generate.py --seeds=0-1000
--network=output/00004-custom_datasets-auto1/network-snapshot-002800.pkl --outdir=result
2022-05-14 19:59:40.319154: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0

评估模型

python calc_metrics.py
--network=output/00004-custom_datasets-auto1/network-snapshot-002800.pkl --metrics=is50k,ppl_wend --metricdata=datasets/custom_datasets/
使用pip install torch-fidelity进行评估
fidelity --gpu 0 --fid --input1 2800result/ --input2 dataset/
fidelity --gpu 0 --isc --input1 1200result/
fidelity --gpu 0 --kid --input1 2800result/ --input2 dataset/

最后

以上就是悦耳小猫咪为你收集整理的stylegan训练的全部内容，希望文章能够帮你解决stylegan训练所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。