本文提供应对课设,毕设的小伙伴们一些参考:项目是参照网络上比较火的一个项目AnimeGAN V2,然后当时的一个课题也是跟这个相关,然后我是直接调用这个的,但是中间遇到过很多的问题,现在想说明一些问题;
项目地址:
https://github.com/bryandlee/animegan2-pytorch
一.安装软件
(1)需要用到pycharm,首先需要安装好,官网为:PyCharm: the Python IDE for Professional Developers by JetBrains
选择对应的版本进行下载;
(2)安装anacoda这个软件,官网为:
Anaconda | The World's Most Popular Data Science Platform
大家可以根据下面网址进行操作,B站上的:
1. Pytorch零失败安装使用|Anaconda安装|最新版本torch安装|PyCharm配置torch| 保姆级教程【PyTorch修炼】【AI蜗牛车】_哔哩哔哩_bilibili里面可能有一些问题,第一个是环境变量最好手动配置,第一次自动安装发现没有安装到环境变量上去,不知道为什么?第二个是anaconda的版本最好下载3.8版本的python,因为可能最新的会导致最后适配出错,安装好后就安装Pytorch,我选的是cpu版本的(因为电脑没有cuda),然后根据视频的操作来,虽然中间会有各种各样的错误,比如说安装失败,多半是网络问题,第三个是视频中的清华镜像好像不能使用了,建议大家直接下载或者挂梯子,我自己下载的话好像很快;
(3)就是适配整个程序,参考博客:AnimeGANv2:照片动漫化_周先森爱吃素的博客-CSDN博客
这位大神是训练出自己的模型,不用装TensorFlow框架好像,一开始我是尝试了一下,然后报错pytorch要1.71以上,重新装了一遍,后来发现不用该博主的训练的更好一些,直接调用v2原型,不用装这个TensorFlow都可以运行;但是要按照该博主的步骤来(除了网盘连接那一部分)
按照下面的步骤来,具体看原博主怎么操作的
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
然后:python test.py --checkpoint [模型文件路径] --input_dir [输入图像所在目录] --output_dir [输出目录] --device [设备选择,cpu或者cuda]
(4)我的话是直接调用v2的文件,没安装TensorFlow就直接可以用,也不知道为什么,以下是我的步骤
打开软件
- 找到文件路径
调用test.py文件
调用的代码参考(3)中的链接中的,里面有介绍
- 显示图片
可以显示出很多图片,但是只展示了一张,大家可以看看,效果还是挺好的,以上就是全部的内容,配环境还是会出现很多问题,有一些问题大家直接百度,基本都有解答,但是可能会有不同的解决方法,大家多尝试一下,个人感觉还是比较容易的,毕竟之前有大佬给过步骤了,我只是完善了一下,新人博主,请大家多多关照,可能在5月弄懂相应的代码和算法,谢谢大家;