Update README.md

README.md

@@ -1,12 +1,106 @@
 ---
 title: MimicMotion
-emoji: 😻
+emoji: 🤸‍♀️
 colorFrom: blue
 colorTo: purple
 sdk: gradio
 sdk_version: 4.39.0
 app_file: app.py
 pinned: false
+suggested_hardware: a10g-large
 ---
+# MimicMotion
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+<a href='http://tencent.github.io/MimicMotion'><img src='https://img.shields.io/badge/Project-Page-Green'></a> <a href='https://arxiv.org/abs/2406.19680'><img src='https://img.shields.io/badge/Paper-Arxiv-red'></a> [![Replicate](https://replicate.com/zsxkib/mimic-motion/badge)](https://replicate.com/zsxkib/mimic-motion)
+
+MimicMotion: High-Quality Human Motion Video Generation with Confidence-aware Pose Guidance
+<br/>
+*Yuang Zhang<sup>1,2</sup>, Jiaxi Gu<sup>1</sup>, Li-Wen Wang<sup>1</sup>, Han Wang<sup>1,2</sup>, Junqi Cheng<sup>1</sup>, Yuefeng Zhu<sup>1</sup>, Fangyuan Zou<sup>1</sup>*
+<br/>
+[<sup>1</sup>Tencent; <sup>2</sup>Shanghai Jiao Tong University]
+
+<p align="center">
+  <img src="assets/figures/preview_1.gif" width="100" />
+  <img src="assets/figures/preview_2.gif" width="100" />
+  <img src="assets/figures/preview_3.gif" width="100" />
+  <img src="assets/figures/preview_4.gif" width="100" />
+  <img src="assets/figures/preview_5.gif" width="100" />
+  <img src="assets/figures/preview_6.gif" width="100" />
+  <br/>
+  <span>Highlights: <b>rich details</b>, <b> good temporal smoothness</b>, and <b>long video length</b>. </span>
+</p>
+
+## Overview
+
+<p align="center">
+  <img src="assets/figures/model_structure.png" alt="model architecture" width="640"/>
+  </br>
+  <i>An overview of the framework of MimicMotion.</i>
+</p>
+
+In recent years, generative artificial intelligence has achieved significant advancements in the field of image generation, spawning a variety of applications. However, video generation still faces considerable challenges in various aspects such as controllability, video length, and richness of details, which hinder the application and popularization of this technology. In this work, we propose a controllable video generation framework, dubbed *MimicMotion*, which can generate high-quality videos of arbitrary length with any motion guidance. Comparing with previous methods, our approach has several highlights. Firstly, with confidence-aware pose guidance, temporal smoothness can be achieved so model robustness can be enhanced with large-scale training data. Secondly, regional loss amplification based on pose confidence significantly eases the distortion of image significantly. Lastly, for generating long smooth videos, a progressive latent fusion strategy is proposed. By this means, videos of arbitrary length can be generated with acceptable resource consumption. With extensive experiments and user studies, MimicMotion demonstrates significant improvements over previous approaches in multiple aspects.
+
+## Quickstart
+
+### Environment setup
+
+Recommend python 3+ with torch 2.x are validated with an Nvidia V100 GPU. Follow the command below to install all the dependencies of python:
+
+```
+conda env create -f environment.yaml
+conda activate mimicmotion
+```
+
+### Download weights
+Please download weights manually as follows:
+```
+cd MimicMotions/
+mkdir models
+```
+1. Download SVD model: [stabilityai/stable-video-diffusion-img2vid-xt-1-1](https://huggingface.co/stabilityai/stable-video-diffusion-img2vid-xt-1-1)
+    ```
+    git lfs install
+    git clone https://huggingface.co/stabilityai/stable-video-diffusion-img2vid-xt-1-1
+    mkdir -p models/SVD
+    mv stable-video-diffusion-img2vid-xt-1-1 models/SVD/
+    ```
+2. Download DWPose pretrained model: [dwpose](https://huggingface.co/yzd-v/DWPose/tree/main)
+    ```
+    git lfs install
+    git clone https://huggingface.co/yzd-v/DWPose
+    mv DWPose models/
+    ```
+3. Download the pre-trained checkpoint of MimicMotion from [Huggingface](https://huggingface.co/ixaac/MimicMotion)
+    ```
+    curl -o models/MimicMotion.pth https://huggingface.co/ixaac/MimicMotion/resolve/main/MimicMotion.pth
+    ```
+
+Finally, all the weights should be organized in models as follows
+
+```
+models/
+├── DWPose
+│   ├── dw-ll_ucoco_384.onnx
+│   └── yolox_l.onnx
+├── SVD
+│   └──stable-video-diffusion-img2vid-xt-1-1
+└── MimicMotion.pth
+```
+
+### Model inference
+
+We provide the inference script.
+```
+python inference.py --inference_config configs/test.yaml
+```
+
+
+## Citation	
+```bib
+@article{mimicmotion2024,
+  title={MimicMotion: High-Quality Human Motion Video Generation with Confidence-aware Pose Guidance},
+  author={Yuang Zhang and Jiaxi Gu and Li-Wen Wang and Han Wang and Junqi Cheng and Yuefeng Zhu and Fangyuan Zou},
+  journal={arXiv preprint arXiv:2406.19680},
+  year={2024}
+}
+```