trrahul / densepose-video

Code to run densepose on video with detectron. https://github.com/facebookresearch/Detectron
GNU General Public License v3.0
62 stars 18 forks source link

Densepose on video

Put the .py files with your detectron installation.

Use command
python2 tools/infer_vid.py     
--cfg configs/DensePose_ResNet101_FPN_s1x-e2e.yaml     
--output-dir DensePoseData/infer_out/    
--wts https://s3.amazonaws.com/densepose/DensePose_ResNet101_FPN_s1x-e2e.pkl 
--input-file filename

Note: Requires about 4GB VRAM.

If you use ResNet101_FPN_s1x model, you need more than 4GB of memory. If you have only 4GB use ResNet50_FPN_s1x.

You can get configs and models from here.

Additional requirement - ffmpeg

Tested with
Example, ran on Geralt doing some witcherin'

examples/geralt.gif

And a bit of Night city.

examples/cpunk.gif