如何用paddlegan进行训练

fpyz

已于 2023-04-02 11:00:02 修改

阅读量355

点赞数

文章标签： python 开发语言

于 2023-03-04 18:57:48 首次发布

本文链接：https://blog.csdn.net/weixin_50864211/article/details/129332139

版权

本文介绍了如何使用PaddleGAN进行训练，包括准备工作、核心配置文件的设置，以及在遇到MemoryError时的处理。提供了AnimeGANPredictor的使用示例，并强调了PaddleGAN的文档阅读和参数调整的重要性，探讨了数据集的选择和训练策略。

摘要由CSDN通过智能技术生成

本文做一个使用paddlegan进行训练的记录，之前aistudio上能看到的都是直接调用现成api的，如果本文有什么不清楚的，欢迎评论区交流

aistudio链接：
https://aistudio.baidu.com/aistudio/projectdetail/5381981

ps:晕，之前项目未公开，挂了这么久，之后才发现

准备工作

核心配置文件

https://github.com/PaddlePaddle/PaddleGAN/blob/develop/docs/zh_CN/install.md

安装apddlepaddle

安装daddlegan

已经训练好的图片进行预测

PATH_OF_IMAGE=r'/home/aistudio/data/zjl.jpg'

from PaddleGAN.ppgan.apps import AnimeGANPredictor

#PATH_OF_IMAGE=r'/home/aistudio/data/jielun.jpg'
PATH_OF_IMAGE=r'/home/aistudio/data/zjl.jpg'
PATH_OF_WEIGHT='/home/aistudio/work/PaddleGAN/output_dir/animeganv2-2023-02-05-12-59/epoch_30_checkpoint.pdparams'
# 直接调用模型
print(PATH_OF_IMAGE)
from ppgan.apps import AnimeGANPredictor
predictor = AnimeGANPredictor(weight_path='/home/aistudio/work/Paddl