Skip to content

trrahul/densepose-video

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

24 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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
  • Ubuntu 16.04
  • Cuda 9.0
  • CUDNN 6
  • opencv 3.4
  • GTX 1050 Ti
  • GTX 1080 Ti
Example, ran on Geralt doing some witcherin'

examples/geralt.gif

And a bit of Night city.

examples/cpunk.gif

Releases

No releases published

Packages

No packages published

Languages