首页 > 其他 > 详细

在矩池云上复现 PaddleGAN 照片转油画风格教程

时间:2021-08-24 11:31:23      阅读:53      评论:0      收藏:0      [点我收藏+]

技术分享图片

我选用的是cuda10.1纯镜像

技术分享图片

案例来自paddlepaddle官方PaddleGAN库中,可以查看

https://github.com/PaddlePaddle/PaddleGAN/

https://github.com/wzmsltw/PaintTransformer

安装需要的软件包

apt-get install cuda-cusolver-dev-10-1

conda install paddlepaddle-gpu==2.1.2 cudatoolkit=10.1 --channel https://mirrors.bfsu.edu.cn/anaconda/cloud/Paddle/

conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.1 -c pytorch

一键上手

拉取并试用

git clone https://gitee.com/matpools/PaintTransformer.git
cd PaintTransformer/inference/
python inference.py

然后在inference.py文件中修改成你自己图片的名字,保存运行即可。

技术分享图片
技术分享图片

完成。

自己配置

也可以使用拉取官方镜像

git clone https://github.com/wzmsltw/PaintTransformer.git

下载mode,可以通过官方下载链接自己下载,然后上传到网盘。

mv /mnt/paint_best.pdparams PaintTransformer/inference/
cd PaintTransformer/inference
python inference.py

检查显存使用率

nvidia-smi -l 5

技术分享图片

参考文章

https://www.paddlepaddle.org.cn/install/quick?docurl=/documentation/docs/zh/install/conda/linux-conda.html

https://github.com/wzmsltw/PaintTransformer

https://github.com/PaddlePaddle/PaddleGAN/blob/develop/README_cn.md

在矩池云上复现 PaddleGAN 照片转油画风格教程

原文:https://www.cnblogs.com/matpool/p/15179094.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!