1.首先需要一个python版本为3.7的conda环境(3.6安装apex会报错)
2.按照readme将训练Celebahq (mask-to-face)所需的文件以及依赖都安装好
- cocosnet代码
- 数据集使用CelebAMask-HQ,作者提供了百度云链接。
- 其它文件我分享到了百度云里。
3.安装apex
- 克隆apex
- 执行
cd apex
- 执行
pip3 install -v --no-cache-dir ./
4.按照自己的机器配置和readme编辑对应的执行命令,运行之后可能会出现维度对不上的BUG,根据issues的回答就可以解决。
Thanks for your reply. For me, the problem is solved by setting the
self.preprocess_input(data, ) to self.preprocess_input(data.copy(), )
in line 52 of pix2pix_model.py.
5.最后在运行的时候可能还会遇到/lib/x86_64-linux-gnu/libstdc++.so.6: version GLIBCXX_3.4.29' not found
的问题。
- 可以尝试搜索本机上所有的libstdc++.so.6
sudo find / -name "libstdc++.so.6*"
。 - 利用
strings path | grep GLIBC
(path为libstdc++.so.6路径),检查哪一个libstdc++.so.6有所需要的 GLIBCXX。 - 最后执行
sudo cp path /usr/lib/x86_64-linux-gnu/libstdc++.so.6
把原来的文件覆盖掉,记得最好先把原来的libstdc++.so.6备份一下,以免出问题。