项目名:video-object-removal
中文名:从视频里“擦除”框选的目标
作者:zllrunning
来源:Fast Online Object Tracking and Segmentation: A Unifying Approach》《Deep Video Inpainting》(CVPR 2019)
论文链接http://openaccess.thecvf.com/content_CVPR_2019/supplemental/Wang_Fast_Online_Object_CVPR_2019_supplemental.pdf
简介:只需绘制一个边界框,你就可以擦除想要删除的对象。所有代码都已经在Ubuntu 16.04,Python 3.5,Pytorch 0.4.0,CUDA 8.0,GTX1080Ti GPU上进行了测试。
Github:https://github.com/zllrunning/video-object-removal
效果:
原图太大,无法上传
https://github.com/zllrunning/video-object-removal/blob/master/results/sgif.gif
安装
git clone https://github.com/zllrunning/video-object-removal.git

cd video-object-removal

cd get_mask

bash make.sh

cd ../inpainting

bash install.sh

cd ..


Demo

1# Download pretrained models of SiamMask and Inpainting
2# Put them in cp/ folder
3# Then just run:
4
python demo.py --data data/Human6

5
6# It also supports video file.
7
python demo.py --data data/bag.avi

8
9# Another optional parameter : --mask-dilation
10
python demo.py --data data/Human6  --mask-dilation 
24
11# This parameter controls the size of the dilation kernel used for the mask. The role is to expand the range of the mask to avoid edge problems. Please see inpainting/davis.py for more details.
推荐阅读
喜欢就点击“在看”吧!
继续阅读
阅读原文