尝试在Windows 10下使用SkyAR更换天空

最近了解到这个有趣的项目,遂尝试自己部署并记录下其中遇到的一些坑。

项目地址:https://github.com/jiupinjia/SkyAR

训练模型:https://drive.google.com/file/d/1COMROzwR4R_7mym6DL9LXhHQlJmJaV0J/view?usp=sharing


首先clone项目到本地,并下载模型文件。将模型带目录解压到项目目录内。

安装Python3, Anaconda:https://www.anaconda.com/products/individual#windows

安装项目目录内Requirements.txt内的依赖,但有几个需要注意/单独安装:

  • pytorch 使用conda安装,11.0是cuda的版本:
    conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch
  • numpy 只能安装1.19.3,否则会报错:
    pip3 install numpy==1.19.3
  • 不装opencv-python,只装opencv-contrib-python:
    pip3 install opencv-contrib-python

安装完成后执行

python .\skymagic.py --path .\config\config-canyon-district9ship.json
即可看到效果。

附一个自己拍的视频,效果貌似不是很好…?

《尝试在Windows 10下使用SkyAR更换天空》上有1条评论

发表评论

电子邮件地址不会被公开。 必填项已用*标注