前言
浅浅记录一下运行这个代码时遇到的坑
一、下载代码和预训练模型
https://github.com/12dmodel/deep_motion_mag
https://people.csail.mit.edu/tiam/deepmag/data.zip
二、在VM虚拟机上安装Ubuntu
版本:Ubuntu 16.04.7
使用VMware Tools传输文件
三、安装环境
打开终端
1、首先安装python-dev:
sudo apt-get install python-dev
2、若你还未安装pip可用此命令安装:sudo apt-get install python-pip,若已安装pip则直接安装以下环境:
pip install configobj==5.0.6
pip install numpy==1.14.2
pip install opencv-python==3.4.2.17
pip install scipy==1.1.0
pip install setproctitle==1.1.10
pip install tensorflow==1.3.0
pip install tqdm==4.23.4
四、程序运行
1、将预训练模型data文件夹放置在deep_motion_mag-master文件夹下
2、打开deep_motion_mag-master的magnet.py,将acvonv更换为ffmpeg
3、安装ffmpeg
sudo apt-get install ffmpeg
4、将待处理视频命名成baby放在vids文件夹下面
5、右键deep_motion_mag-master文件夹--在终端打开,运行:
ffmpeg -i /home/ubuntu/LBMM/deep_motion_mag-master/data/vids/baby.mp4 -f image2 /home/ubuntu/LBMM/deep_motion_mag-master/data/output/baby/%06d.png
这一步是将视频分帧,运行完后可以在output看到生成图像文件夹baby中有多个.png文件
6、将所生成的整个baby文件夹移动到vids下,运行:
sh run_on_test_videos.sh o3f_hmhm2_bg_qnoise_mix4_nl_n_t_ds3 baby 15 yes
若出现 ImportError: No module named tensorflow,那问题就是上面的库安装在了python3上,而tensorflow不支持python3.7,这时用pip2 install把所有库安装在python2.7上就可以了
运行时间约2小时,耐心等待,结束后可在output文件夹下看到输出视频
总结
下一篇讲怎么训练模型