说明

前几天我接触了一个由2D图片生成3D视频的开源项目:https://github.com/vt-vl-lab/3d-photo-inpainting


当时是在谷歌的Colab平台上运行的,在线平台虽方便,但是连接性不好,时断时续的体验差。最蛋疼的是运行环境不能被保存,第二天想用就得从头安装环境,相当耽误时间。于是我就试了本地部署该项目,虽然官方给出的Linux上的部署教程,但我又不是程序员,懒得整一套Linux系统了,就仿照着在Windows上部署了一下,结果是完全可行的。

前提

  1. 电脑已安装Anaconda
  2. 电脑GPU支持cuda加速并安装相应版本cuda驱动

步骤

  • 打开Anaconda控制台新建一个环境
conda create -n 3DP python=3.7 anaconda
conda activate 3DP
  • 下载项目和模型
git clone git@github.com:vt-vl-lab/3d-photo-inpainting.git

手动下载这里面的4个模型并放置在相应文件夹:https://github.com/vt-vl-lab/3d-photo-inpainting/blob/master/download.sh

这一步我下好并打包放在一起了:https://pan.2bboy.com/Public/Software/www/3d-photo-inpainting.zip

  • 下载依赖库
cd 3d-photo-inpainting
pip install -r requirements.txt
conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit==10.1.243 -c pytorch

下载速度感人请试着换源或代理

  • 将jpg格式的图片放在image目录下,运行命令
python main.py --config argument.yml

更多设置

项目默认运行的参数都写在了argument.yml文件中,这里面的参数意义参考:https://github.com/vt-vl-lab/3d-photo-inpainting/blob/master/DOCUMENTATION.md
这里面参数可以控制输出视频大小和镜头运动等,具体设置就自己探索吧。

Last modification:April 28th, 2020 at 02:43 pm