最近了解到这个有趣的项目,遂尝试自己部署并记录下其中遇到的一些坑。
项目地址: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
即可看到效果。
附一个自己拍的视频,效果貌似不是很好…?
. .