1.GitHub上有PSGAN的源码
注意还要在ReadMe的Train里头下载训练数据集data。链接:
GitHub - wtjiang98/PSGAN: PyTorch code for "PSGAN: Pose and Expression Robust Spatial-Aware GAN for Customizable Makeup Transfer" (CVPR 2020 Oral)
尝试跑通。本来一直在我的无独显的小笔记本上折腾,然后一直报错,不信邪的折腾了好几天,直到在舍友的3060上试了一下,比我几天的进展还好。大四现在不论买新游戏本还是在宿舍组台式实在都太不划算,开始找云服务器。
2.租云服务器
看网上推荐的有colab,kaggle,九天毕昇,autodl等,前俩貌似要谷歌账号,九天非常不稳定我注册的时候都卡,最后选择了autodl。
官网:AutoDL算力云 | 弹性、好用、省钱。租GPU就上AutoDL
点进去注册之后,充点钱,就能进算力市场租显卡了。这个平台的缺点是人多货少,还得靠抢。
3.配置镜像
Requirements里面会有推荐配置的环境(当然不按推荐也问题不大)。
选择基础镜像,选好你想要的配置,然后立即创建:
等待创建完成,在控制台就会出现刚才创建的实例啦~
4.导入项目
实例创建成功后,点击 开机 -- 关机 ,在 更多 里选择 无卡模式开机(因为无卡模式便宜哇)。
等待开机之后,点击JupyterLab,就能把代码数据之类的传到这个实例啦。
点进去之后长这个样:
我看的教程说的是把项目传到tmp那个文件夹里,据说那个文件夹里读写比较快🤔
打开tmp那个文件夹后,直接拖拽上传就行。由于不支持上传文件夹,可以把项目打包压缩为一个zip压缩包再上传。等待上传结束,打开终端,使用unzip解压就行啦。(tar啥的也行,就是解压命令不一样,自己搜。)
解压完这个样子,tmp文件夹里多了项目的文件夹:
5.运行项目
在终端中cd进入PSGAN-master的文件夹,然后输入命令运行train.py文件:
cd autodl-tmp/PSGAN-master
python train.py
然后会报好几次错,看error信息下载下载对应的包就行。
我记得有缺少cv2的错误,命令:pip install opencv-python
缺少dlib的错误,这个我忘记用的啥命令了,可能是下面这个也可能不是:
conda install -c https://conda.anaconda.org/conda-forge dlib
还有个错误,下面这个教程成功解决:
当报错ZeroDivisionError: integer division or modulo by zero的时候说明环境配好啦,这个错误是因为没有GPU。
然后我们要关机,开机,这个时候要用有GPU的,假如没空闲的要等等。
进入之后,在终端重复命令运行train.py:
cd autodl-tmp/PSGAN-master
python train.py
然后就运行成功啦(至少我应该是运行成功了)。
项目下会多一个叫log的文件夹,里面有一些看不懂的文件,以后我再研究。
就是下面这些:
就酱,我也不造运行成功没,反正没报错就是胜利,具体代码明天再看,完毕。
对了!不用的时候一定要记得关机!!!不关机就在扣钱!!!
这个状态是已关机就行了。
over。