camenduru commited on
Commit
20236ef
·
verified ·
1 Parent(s): 4c079e5

thanks to BadToBest ❤

Browse files
Files changed (5) hide show
  1. README.md +247 -0
  2. denoising_unet.pth +3 -0
  3. motion_module.pth +3 -0
  4. pose_encoder.pth +3 -0
  5. reference_unet.pth +3 -0
README.md ADDED
@@ -0,0 +1,247 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ <h1 align='center'>EchoMimicV2: Towards Striking, Simplified, and Semi-Body Human Animation</h1>
2
+
3
+ <div align='center'>
4
+ <a href='https://github.com/mengrang' target='_blank'>Rang Meng</a><sup></sup>&emsp;
5
+ <a href='https://github.com/' target='_blank'>Xingyu Zhang</a><sup></sup>&emsp;
6
+ <a href='https://lymhust.github.io/' target='_blank'>Yuming Li</a><sup></sup>&emsp;
7
+ <a href='https://github.com/' target='_blank'>Chenguang Ma</a><sup></sup>
8
+ </div>
9
+
10
+
11
+ <div align='center'>
12
+ Terminal Technology Department, Alipay, Ant Group.
13
+ </div>
14
+ <br>
15
+ <div align='center'>
16
+ <a href='https://antgroup.github.io/ai/echomimic_v2/'><img src='https://img.shields.io/badge/Project-Page-blue'></a>
17
+ <a href='https://huggingface.co/BadToBest/EchoMimicV2'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20HuggingFace-Model-yellow'></a>
18
+ <!--<a href='https://antgroup.github.io/ai/echomimic_v2/'><img src='https://img.shields.io/badge/%F0%9F%A4%97%20HuggingFace-Demo-yellow'></a>-->
19
+ <a href='https://huggingface.co/BadToBest/EchoMimicV2'><img src='https://img.shields.io/badge/ModelScope-Model-purple'></a>
20
+ <!--<a href='https://antgroup.github.io/ai/echomimic_v2/'><img src='https://img.shields.io/badge/ModelScope-Demo-purple'></a>-->
21
+ <a href='https://arxiv.org/abs/2411.10061'><img src='https://img.shields.io/badge/Paper-Arxiv-red'></a>
22
+ <a href='https://github.com/antgroup/echomimic_v2/blob/main/assets/halfbody_demo/wechat_group.png'><img src='https://badges.aleen42.com/src/wechat.svg'></a>
23
+ </div>
24
+
25
+ ## &#x1F680; EchoMimic Series
26
+ * EchoMimicV1: Lifelike Audio-Driven Portrait Animations through Editable Landmark Conditioning. [GitHub](https://github.com/antgroup/echomimic)
27
+ * EchoMimicV2: Towards Striking, Simplified, and Semi-Body Human Animation. [GitHub](https://github.com/antgroup/echomimic_v2)
28
+
29
+ ## &#x1F4E3; Updates
30
+ * [2024.11.19] 🔥 We release the EMTD dataset list and processing scripts.
31
+ * [2024.11.19] 🔥 We release our [EchoMimicV2](https://github.com/antgroup/echomimic_v2) codes and models.
32
+ * [2024.11.15] 🔥 Our [paper](https://arxiv.org/abs/2411.10061) is in public on arxiv.
33
+
34
+ ## &#x1F305; Gallery
35
+ ### Introduction
36
+ <table class="center">
37
+ <tr>
38
+ <td width=50% style="border: none">
39
+ <video controls loop src="https://github.com/user-attachments/assets/f544dfc0-7d1a-4c2c-83c0-608f28ffda25" muted="false"></video>
40
+ </td>
41
+ <td width=50% style="border: none">
42
+ <video controls loop src="https://github.com/user-attachments/assets/7f626b65-725c-4158-a96b-062539874c63" muted="false"></video>
43
+ </td>
44
+ </tr>
45
+ </table>
46
+
47
+ ### English Driven Audio
48
+ <table class="center">
49
+ <tr>
50
+ <td width=100% style="border: none">
51
+ <video controls loop src="https://github.com/user-attachments/assets/3d5ac52c-62e4-41bc-8b27-96f005bbd781" muted="false"></video>
52
+ </td>
53
+ </tr>
54
+ </table>
55
+ <table class="center">
56
+ <tr>
57
+ <td width=30% style="border: none">
58
+ <video controls loop src="https://github.com/user-attachments/assets/e8dd6919-665e-4343-931f-54c93dc49a7d" muted="false"></video>
59
+ </td>
60
+ <td width=30% style="border: none">
61
+ <video controls loop src="https://github.com/user-attachments/assets/2a377391-a0d3-4a9d-8dde-cc59006e7e5b" muted="false"></video>
62
+ </td>
63
+ <td width=30% style="border: none">
64
+ <video controls loop src="https://github.com/user-attachments/assets/462bf3bb-0af2-43e2-a2dc-559e79953f3c" muted="false"></video>
65
+ </td>
66
+ </tr>
67
+ <tr>
68
+ <td width=30% style="border: none">
69
+ <video controls loop src="https://github.com/user-attachments/assets/0e988e7f-6346-4b54-9061-9cfc7a80e9c8" muted="false"></video>
70
+ </td>
71
+ <td width=30% style="border: none">
72
+ <video controls loop src="https://github.com/user-attachments/assets/56f739bd-afbf-4ed3-ab15-73a811c1bc46" muted="false"></video>
73
+ </td>
74
+ <td width=30% style="border: none">
75
+ <video controls loop src="https://github.com/user-attachments/assets/1b2f7827-111d-4fc0-a773-e1731bba285d" muted="false"></video>
76
+ </td>
77
+ </tr>
78
+ <tr>
79
+ <td width=30% style="border: none">
80
+ <video controls loop src="https://github.com/user-attachments/assets/a76b6cc8-89b9-4f7e-b1ce-c85a657b6dc7" muted="false"></video>
81
+ </td>
82
+ <td width=30% style="border: none">
83
+ <video controls loop src="https://github.com/user-attachments/assets/bf03b407-5033-4a30-aa59-b8680a515181" muted="false"></video>
84
+ </td>
85
+ <td width=30% style="border: none">
86
+ <video controls loop src="https://github.com/user-attachments/assets/f98b3985-572c-499f-ae1a-1b9befe3086f" muted="false"></video>
87
+ </td>
88
+ </tr>
89
+ </table>
90
+
91
+ ### Chinese Driven Audio
92
+ <table class="center">
93
+ <tr>
94
+ <td width=30% style="border: none">
95
+ <video controls loop src="https://github.com/user-attachments/assets/a940a332-2fd1-48e7-b3c4-f88f63fd1c9d" muted="false"></video>
96
+ </td>
97
+ <td width=30% style="border: none">
98
+ <video controls loop src="https://github.com/user-attachments/assets/8f185829-c67f-45f4-846c-fcbe012c3acf" muted="false"></video>
99
+ </td>
100
+ <td width=30% style="border: none">
101
+ <video controls loop src="https://github.com/user-attachments/assets/a49ab9be-f17b-41c5-96dd-20dc8d759b45" muted="false"></video>
102
+ </td>
103
+ </tr>
104
+ <tr>
105
+ <td width=30% style="border: none">
106
+ <video controls loop src="https://github.com/user-attachments/assets/1136ec68-a13c-4ee7-ab31-5621530bf9df" muted="false"></video>
107
+ </td>
108
+ <td width=30% style="border: none">
109
+ <video controls loop src="https://github.com/user-attachments/assets/fc16d512-8806-4662-ae07-8fcf45c75a83" muted="false"></video>
110
+ </td>
111
+ <td width=30% style="border: none">
112
+ <video controls loop src="https://github.com/user-attachments/assets/f8559cd1-f555-4781-9251-dfcef10b5b01" muted="false"></video>
113
+ </td>
114
+ </tr>
115
+ <tr>
116
+ <td width=30% style="border: none">
117
+ <video controls loop src="https://github.com/user-attachments/assets/c7473e3a-ab51-4ad5-be96-6c4691fc0c6e" muted="false"></video>
118
+ </td>
119
+ <td width=30% style="border: none">
120
+ <video controls loop src="https://github.com/user-attachments/assets/ca69eac0-5126-41ee-8cac-c9722004d771" muted="false"></video>
121
+ </td>
122
+ <td width=30% style="border: none">
123
+ <video controls loop src="https://github.com/user-attachments/assets/e66f1712-b66d-46b5-8bbd-811fbcfea4fd" muted="false"></video>
124
+ </td>
125
+ </tr>
126
+ </table>
127
+
128
+ ## ⚒️ Installation
129
+ ### Download the Codes
130
+
131
+ ```bash
132
+ git clone https://github.com/antgroup/echomimic_v2
133
+ cd echomimic_v2
134
+ ```
135
+
136
+ ### Python Environment Setup
137
+
138
+ - Tested System Environment: Centos 7.2/Ubuntu 22.04, Cuda >= 11.7
139
+ - Tested GPUs: A100(80G) / RTX4090D (24G) / V100(16G)
140
+ - Tested Python Version: 3.8 / 3.10 / 3.11
141
+
142
+ Create conda environment (Recommended):
143
+
144
+ ```bash
145
+ conda create -n echomimic python=3.8
146
+ conda activate echomimic
147
+ ```
148
+
149
+ Install packages with `pip`
150
+ ```bash
151
+ pip install -r requirements.txt
152
+ ```
153
+
154
+ ### Download ffmpeg-static
155
+ Download and decompress [ffmpeg-static](https://www.johnvansickle.com/ffmpeg/old-releases/ffmpeg-4.4-amd64-static.tar.xz), then
156
+ ```
157
+ export FFMPEG_PATH=/path/to/ffmpeg-4.4-amd64-static
158
+ ```
159
+
160
+ ### Download pretrained weights
161
+
162
+ ```shell
163
+ git lfs install
164
+ git clone https://huggingface.co/BadToBest/EchoMimicV2 pretrained_weights
165
+ ```
166
+
167
+ The **pretrained_weights** is organized as follows.
168
+
169
+ ```
170
+ ./pretrained_weights/
171
+ ├── denoising_unet.pth
172
+ ├── reference_unet.pth
173
+ ├── motion_module.pth
174
+ ├── pose_encoder.pth
175
+ ├── sd-vae-ft-mse
176
+ │ └── ...
177
+ ├── sd-image-variations-diffusers
178
+ │ └── ...
179
+ └── audio_processor
180
+ └── whisper_tiny.pt
181
+ ```
182
+
183
+ In which **denoising_unet.pth** / **reference_unet.pth** / **motion_module.pth** / **face_locator.pth** are the main checkpoints of **EchoMimic**. Other models in this hub can be also downloaded from it's original hub, thanks to their brilliant works:
184
+ - [sd-vae-ft-mse](https://huggingface.co/stabilityai/sd-vae-ft-mse)
185
+ - [sd-image-variations-diffusers](https://huggingface.co/lambdalabs/sd-image-variations-diffusers)
186
+ - [audio_processor(whisper)](https://openaipublic.azureedge.net/main/whisper/models/65147644a518d12f04e32d6f3b26facc3f8dd46e5390956a9424a650c0ce22b9/tiny.pt)
187
+
188
+ ### Inference on Demo
189
+ Run the python inference script:
190
+ ```bash
191
+ python infer.py --config='./configs/prompts/infer.yaml'
192
+ ```
193
+
194
+ ### EMTD Dataset
195
+ Download dataset:
196
+ ```bash
197
+ python ./EMTD_dataset/download.py
198
+ ```
199
+ Slice dataset:
200
+ ```bash
201
+ bash ./EMTD_dataset/slice.sh
202
+ ```
203
+ Process dataset:
204
+ ```bash
205
+ python ./EMTD_dataset/preprocess.py
206
+ ```
207
+
208
+ ## 📝 Release Plans
209
+
210
+ | Status | Milestone | ETA |
211
+ |:--------:|:-------------------------------------------------------------------------|:--:|
212
+ | ✅ | The inference source code of EchoMimicV2 meet everyone on GitHub | 21st Nov, 2024 |
213
+ | ✅ | Pretrained models trained on English and Mandarin Chinese on HuggingFace | 21st Nov, 2024 |
214
+ | ✅ | Pretrained models trained on English and Mandarin Chinese on ModelScope | 21st Nov, 2024 |
215
+ | ✅ | EMTD dataset list and processing scripts | 21st Nov, 2024 |
216
+ | 🚀 | Accelerated models to be released | TBD |
217
+ | 🚀 | Online Demo on ModelScope to be released | TBD |
218
+ | 🚀 | Online Demo on HuggingFace to be released | TBD |
219
+
220
+ ## ⚖️ Disclaimer
221
+ This project is intended for academic research, and we explicitly disclaim any responsibility for user-generated content. Users are solely liable for their actions while using the generative model. The project contributors have no legal affiliation with, nor accountability for, users' behaviors. It is imperative to use the generative model responsibly, adhering to both ethical and legal standards.
222
+
223
+ ## 🙏🏻 Acknowledgements
224
+
225
+ We would like to thank the contributors to the [MimicMotion](https://github.com/Tencent/MimicMotion) and [Moore-AnimateAnyone](https://github.com/MooreThreads/Moore-AnimateAnyone) repositories, for their open research and exploration.
226
+
227
+ We are also grateful to [CyberHost](https://cyberhost.github.io/) and [Vlogger](https://enriccorona.github.io/vlogger/) for their outstanding work in the area of audio-driven human animation.
228
+
229
+ If we missed any open-source projects or related articles, we would like to complement the acknowledgement of this specific work immediately.
230
+
231
+ ## &#x1F4D2; Citation
232
+
233
+ If you find our work useful for your research, please consider citing the paper :
234
+
235
+ ```
236
+ @misc{meng2024echomimic,
237
+ title={EchoMimicV2: Towards Striking, Simplified, and Semi-Body Human Animation},
238
+ author={Rang Meng, Xingyu Zhang, Yuming Li, Chenguang Ma},
239
+ year={2024},
240
+ eprint={2411.10061},
241
+ archivePrefix={arXiv},
242
+ primaryClass={cs.CV}
243
+ }
244
+ ```
245
+
246
+ ## &#x1F31F; Star History
247
+ [![Star History Chart](https://api.star-history.com/svg?repos=antgroup/echomimic_v2&type=Date)](https://star-history.com/#antgroup/echomimic_v2&Date)
denoising_unet.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a4417d929ad13f02d8e6a7607aa2e3b147451e131a52bcb499b2c992ad01c327
3
+ size 1702192597
motion_module.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:15d5d2bb5d184eaf9475285f0d5068bde32ae70c3961c799559ee3a6a25afabd
3
+ size 908951917
pose_encoder.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:24113dffd754519d67ef960cb30a5e53958bb552d7ca7cafcd94addbe840204c
3
+ size 1702048079
reference_unet.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cfc1d7a28bd33daf365470d21a318833a014c43624314689eef6cc5f6a352778
3
+ size 1631280213