search

JackChen890311 / Simple-Magic-Animate

A simple magic animate pipeline including densepose inference.
31 stars 5 forks source link

DensePose Model Download #1

Closed bravma closed 8 months ago

bravma commented 8 months ago

Thanks for putting these two repos together! I was able to run magic-animate using your provided inference-script. However, I have troubles running DensePose. I have downloaded this model here: https://github.com/JackChen890311/Simple-Magic-Animate/blob/main/DensePose/configs/cse/densepose_rcnn_R_101_FPN_DL_s1x.yaml. Is this the one you used as well? Running inference for DensePose seems to work, however, the output looks weird.

image
JackChen890311 commented 8 months ago

Hello,

The densepose model config I used is this one: https://github.com/JackChen890311/Simple-Magic-Animate/blob/main/DensePose/configs/densepose_rcnn_R_101_FPN_DL_s1x.yaml

You should be able to download the checkpoint here: https://github.com/facebookresearch/detectron2/blob/main/projects/DensePose/doc/DENSEPOSE_IUV.md#-improved-baselines-deeplabv3-head

Choose R_101_FPN_DL_s1x under "DeepLabV3 Head" and download the model (or click here).

I'll add this to readme later as well, hope this helps!