说明
前几天我接触了一个由2D图片生成3D视频的开源项目:https://github.com/vt-vl-lab/3d-photo-inpainting
当时是在谷歌的Colab平台上运行的,在线平台虽方便,但是连接性不好,时断时续的体验差。最蛋疼的是运行环境不能被保存,第二天想用就得从头安装环境,相当耽误时间。于是我就试了本地部署该项目,虽然官方给出的Linux上的部署教程,但我又不是程序员,懒得整一套Linux系统了,就仿照着在Windows上部署了一下,结果是完全可行的。
前提
- 电脑已安装Anaconda
- 电脑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
- 下载依赖库
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
这里面参数可以控制输出视频大小和镜头运动等,具体设置就自己探索吧。
博主能不能加加你QQ,指导下我怎么本地部署,该安装的都安装了,其它不会啊,谢谢
打开Anaconda的控制台按照步骤输一遍代码即可,中间需要单独下载几个模型文件而已
https://mailnwpueducn-my.sharepoint.com需要登录才可以下载(下载到50%不让下载了)OωO
请问所有流程都设置成功了,请问下次再想启动的话,怎么搞,不会是按照安装步骤再来一遍吧?
进入环境后,直接运行就行了
大佬你给研究一下动作迁移相关的呗,就视频动作迁移给图片的
仙人快去学习下Tecogan,真的,真的,真的|´・ω・)ノヾ(≧∇≦*)ゝ
dalao,我在运行时出现了CUDA out of memory错误,在网上查找后发现需要调小batch_size,但我第一次接触机器学习,压根不知道这东西在哪里,dalao可否指一条明路?
我只会跑demo,源码我也不太懂
破案了,既然GPU不行那就拿CPU跑,本以为也会报错不过意外的成了