A webui for Propainter

简体中文\ A webui that you can easily pick up objects from the video and eliminate them.

faster-propainter | 提速版

https://github.com/halfzm/faster-propainter

Demo

Getting started

If you don't want to install the environment, you can download the package directly.\ link 百度网盘\ tutorial bilibili youtube

clone repo

git clone git@github.com:halfzm/ProPainter-Webui.git

create conda enviroment

conda create -n propainter python=3.10
conda activate propainter

install dependencies

Just follow the instructions in Segment-ant-Track-Anything 和 ProPainter（P.S.Don't need to install groundingdino, I have put it in the project.）

pip install -r requirements.txt

prepare pretrained models

Download all the needed models for propainter \ propainter\ raft-things\ recurrent_flow_completion\ i3d_rgb_imagenet

Download all the needed models for segment-and-track-anything\ SAM-VIT-B (sam_vit_b_01ec64.pth)\ R50-DeAOT-L (R50_DeAOTL_PRE_YTB_DAV.pth)\ GroundingDINO-T (groundingdino_swint_ogc)\

The directory structure will be arranged as：

ckpt
   |- bert-base-uncased
   |- groundingdino_swint_ogc.pth
   |- R50_EdAOTL_PRE_YTB_DAV.pth
   |- sam_vit_b_01ec64.pth
...
ProPainter/weights
   |- ProPainter.pth
   |- recurrent_flow_completion.pth
   |- raft-things.pth
   |- i3d_rgb_imagenet.pt (for evaluating VFID metric)
   |- README.md

quick start

python app.py

halfzm / ProPainter-Webui

readme