magic-animate README

The project's README from github.com/magic-research/magic-animate. As one markdown file: README.md.

MagicAnimate: Temporally Consistent Human Image Animation using Diffusion Model

Zhongcong Xu · Jianfeng Zhang · Jun Hao Liew · Hanshu Yan · Jia-Wei Liu · Chenxu Zhang · Jiashi Feng · Mike Zheng Shou

Paper PDF Project Page
National University of Singapore   |   ByteDance

📢 News

🏃‍♂️ Getting Started

Download the pretrained base models for StableDiffusion V1.5 and MSE-finetuned VAE.

Download our MagicAnimate checkpoints.

Please follow the huggingface download instructions to download the above models and checkpoints, git lfs is recommended.

Place the based models and checkpoints as follows:

magic-animate
|----pretrained_models
  |----MagicAnimate
    |----appearance_encoder
      |----diffusion_pytorch_model.safetensors
      |----config.json
    |----densepose_controlnet
      |----diffusion_pytorch_model.safetensors
      |----config.json
    |----temporal_attention
      |----temporal_attention.ckpt
  |----sd-vae-ft-mse
    |----config.json
    |----diffusion_pytorch_model.safetensors
  |----stable-diffusion-v1-5
    |----scheduler
       |----scheduler_config.json
    |----text_encoder
       |----config.json
       |----pytorch_model.bin
    |----tokenizer (all)
    |----unet
       |----diffusion_pytorch_model.bin
       |----config.json
    |----v1-5-pruned-emaonly.safetensors
|----...

⚒️ Installation

prerequisites: python>=3.8, CUDA>=11.3, and ffmpeg.

Install with conda:

conda env create -f environment.yaml
conda activate manimate

or pip:

pip3 install -r requirements.txt

💃 Inference

Run inference on single GPU:

bash scripts/animate.sh

Run inference with multiple GPUs:

bash scripts/animate_dist.sh

🎨 Gradio Demo

Online Gradio Demo:

Try our online gradio demo quickly.

Local Gradio Demo:

Launch local gradio demo on single GPU:

python3 -m demo.gradio_animate

Launch local gradio demo if you have multiple GPUs:

python3 -m demo.gradio_animate_dist

Then open gradio demo in local browser.

🙏 Acknowledgements

We would like to thank AK(@_akhaliq) and huggingface team for the help of setting up oneline gradio demo.

🎓 Citation

If you find this codebase useful for your research, please use the following entry.

@inproceedings{xu2023magicanimate,
    author    = {Xu, Zhongcong and Zhang, Jianfeng and Liew, Jun Hao and Yan, Hanshu and Liu, Jia-Wei and Zhang, Chenxu and Feng, Jiashi and Shou, Mike Zheng},
    title     = {MagicAnimate: Temporally Consistent Human Image Animation using Diffusion Model},
    booktitle = {arXiv},
    year      = {2023}
}

magic-animate (magic-research/magic-animate)

Searching Database…

Что это такое?Это официальный код проекта 'MagicAnimate' (CVPR 2024) — метод на основе диффузионных моделей, который анимирует изображение человека-референса по последовательности движений и создаёт видео с согласованностью во времени.
Это готовое приложение или просто исследовательский код?Это официальный релиз исследования к статье: скрипты для инференса и демо на Gradio, построенное на Stable Diffusion V1.5. Это не упакованная библиотека и не установщик.
Как это запустить?Скачайте базовую модель Stable Diffusion V1.5, VAE, дообученный на MSE, и чекпоинты MagicAnimate с Hugging Face, разложите их в нужной структуре папок, а затем запустите предусмотренный скрипт инференса.
Что нужно установить перед запуском?Python 3.8+, CUDA 11.3+ и ffmpeg; git-lfs рекомендуется для загрузки больших предобученных чекпоинтов.
Есть ли подводные камни при установке, о которых стоит знать?Нужно вручную скачать несколько больших предобученных чекпоинтов (базовую модель SD, VAE, веса MagicAnimate) и разложить их в точной структуре папок, прежде чем что-либо запустится.
Чем MagicAnimate отличается от простых методов переноса позы?Он специально нацелен на согласованность во времени между кадрами: использует диффузионную модель, управляемую позой, и уменьшает мерцание и дрейф внешности, характерные для покадровых подходов к анимации.
Можно попробовать без установки?Да, есть онлайн-демо на Gradio (созданное при участии команды Hugging Face), которым можно пользоваться прямо в браузере без локальной настройки.
Это бесплатно, в том числе для коммерческого использования?Да, код распространяется по лицензии BSD-3-Clause, которая является разрешительной и допускает коммерческое использование.
Size
24.9 MB
Stars
10.9K
Updated
CommitAug 29
Age
2y
Commits · 90 Days