Skip to content

janglucky/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

About

Code to run densepose on video with detectron. https://github.com/facebookresearch/Detectron

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%