栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

机器学习——windows上实现照片动漫化【AnimeGANv2】

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

机器学习——windows上实现照片动漫化【AnimeGANv2】

简介

本文介绍的是基于Windows环境下的实现,因为自己是小白也摸索了很久,在此记录步骤。
时间:2022.4.27

项目需求:

1.windows环境

2.anaconda

3.pytorch

前言

机缘巧合,接触到了github上的开源项目:AnimeGANv2 ,作者为:bryandlee

实现效果: 【温馨提示:大家最好使用高清晰度的照片,这样效果更好】

原作者使用的是TensorFlow, 然后看见了周先森爱吃素大佬发表:AnimeGANv2:照片动漫化

 而他是基于Ubuntu实现的,所以在此记录用windows上记录详细步骤。

步骤一:

1.1  下载 项目文件  【里面包含了:项目文件+三个预训练模型文件】

【链接:https://pan.baidu.com/s/1CZUuJivKsFGXP4XWEyyplw 提取码:deng】

modles文件夹中分别是
pytorch_generator_Paprika.pt
pytorch_generator_Shinkai.pt
pytorch_generator_Hayao.pt

分别是:宫崎骏、新海诚和今敏的三种风格

 

步骤二:

2.1  下载anaconda 【建议看视频:简单方便,这里附上b站视频链接】

2.2  换源ananconda 

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/

# 设置搜索时显示通道地址
conda config --set show_channel_urls yes

需要pytorch, 还需要添加pytorch的镜像

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
步骤三:

3.1  打开cmd,切盘到项目文件中【windows下切盘】

进入到项目文件

步骤四:

 4.1  在当前目录下,执行以下命令【conda环境配置】:

conda create -n animegan python=3.8 -y
conda activate animegan
conda install pytorch=1.6 torchvision cudatoolkit=10.2 -c pytorch -y
pip install opencv-python

 现在就差运行代码了【下载时有些久,请耐心等等】

步骤五:

5.1  调用test.py文件——具体的命令格式如下:

python test.py --checkpoint [模型文件路径] --input_dir [输入图像所在目录] --output_dir [输出目录] --device [设备选择,cpu或者cuda]

1.  测试的图片放在根目录的samples下的inputs文件夹中

2.  生成的图片放在根目录的results文件夹中

3.  这里使用CPU进行推理'

4.  使用的是新海诚风格【根据具体情况更改风格】

4.  使用命令:

python test.py --checkpoint ./models/pytorch_generator_Shinkai.pt --input_dir ./samples/inputs/ --output_dir ./results/ --device cpu

附:如果有报错:

 

 请尝试将clip改成clamp

 

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/839747.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号