动态天空置换算法SkyAR体验

动态天空置换算法SkyAR体验

之前在手机拍照过程中体验过一些有趣的AR特效，而近日偶然发现这个动态天空置换算法SkyAR，遂将实现过程记录如下。

github网址：https://github.com/jiupinjia/SkyAR

论文网址：Castle in the Sky: Dynamic Sky Replacement and Harmonization in Videos

效果演示

代码运行

1. 下载代码

git clone https://github.com/jiupinjia/SkyAR.git

2. 搭建虚拟环境

conda create -n SkyAR python=3.9

conda activate SkyAR

3. 安装依赖

conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia

Requirement.txt

matplotlib
scikit-image
scikit-learn
scipy
numpy
torch
torchvision
opencv-python
opencv-contrib-python

4. 下载权重文件

链接：https://pan.baidu.com/s/1gCzZX6Qpxv3PLtcYetuCaw
提取码：gou6

将权重文件放至SkyAr目录下

5. 代码测试

在虚拟环境下运行

python skymagic.py --path ./config/config-canyon-galaxy.json