Upload 24 files
Browse files- uniAnimate/checkpoints/2dfan4.onnx +3 -0
- uniAnimate/checkpoints/Fourchannel_maskChannel_24w_normalize255_best_ckpt.pth +3 -0
- uniAnimate/checkpoints/README.md +41 -0
- uniAnimate/checkpoints/SodAnything_vits_24w_removeRelu_best_ckpt.pth +3 -0
- uniAnimate/checkpoints/arcface_w600k_r50.onnx +3 -0
- uniAnimate/checkpoints/big-lama.pt +3 -0
- uniAnimate/checkpoints/configuration.json +1 -0
- uniAnimate/checkpoints/dw-ll_ucoco_384.onnx +3 -0
- uniAnimate/checkpoints/iic/unianimate/.mdl +0 -0
- uniAnimate/checkpoints/iic/unianimate/.msc +0 -0
- uniAnimate/checkpoints/iic/unianimate/.mv +1 -0
- uniAnimate/checkpoints/inswapper_128.onnx +3 -0
- uniAnimate/checkpoints/models--LiheYoung--depth_anything_vits14/README.md +58 -0
- uniAnimate/checkpoints/models--LiheYoung--depth_anything_vits14/config.json +1 -0
- uniAnimate/checkpoints/models--LiheYoung--depth_anything_vits14/gitattributes +35 -0
- uniAnimate/checkpoints/models--LiheYoung--depth_anything_vits14/pytorch_model.bin +3 -0
- uniAnimate/checkpoints/open_clip_pytorch_model.bin +3 -0
- uniAnimate/checkpoints/portrate_selflabelFFinetune_fullAndCut_dp_size512_ImageANTIALIAS_adduserimage_best_ckpt.pth +3 -0
- uniAnimate/checkpoints/unianimate_16f_32f_non_ema_223000.pth +3 -0
- uniAnimate/checkpoints/v2-1_512-ema-pruned.ckpt +3 -0
- uniAnimate/checkpoints/yoloface_8n.onnx +3 -0
- uniAnimate/checkpoints/yolov8m.pt +3 -0
- uniAnimate/checkpoints/yolov8x.pt +3 -0
- uniAnimate/checkpoints/yolox_l.onnx +3 -0
uniAnimate/checkpoints/2dfan4.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1ceedb108439c7d7b3f92cfa2b25bdc69a1f5f6c8b41da228cb283ca98d4181d
|
| 3 |
+
size 97904402
|
uniAnimate/checkpoints/Fourchannel_maskChannel_24w_normalize255_best_ckpt.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:dcc067fd8771ed997dbee283d2b72ae6b70d059d28c195e119129e15e0ac7b82
|
| 3 |
+
size 110367857
|
uniAnimate/checkpoints/README.md
ADDED
|
@@ -0,0 +1,41 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# UniAnimate: Taming Unified Video Diffusion Models for Consistent Human Image Animation
|
| 2 |
+
|
| 3 |
+
## This repo include the checkpoints for UniAnimate:
|
| 4 |
+
|
| 5 |
+
- "models/dw-ll_ucoco_384.onnx": the checkpoint for dwpose extraction.
|
| 6 |
+
|
| 7 |
+
- "models/open_clip_pytorch_model.bin": the checkpoint for clip embedding.
|
| 8 |
+
|
| 9 |
+
- "models/unianimate_16f_32f_non_ema_223000.pth": the checkpoint for human image animation in UniAnimate (16/32 frames).
|
| 10 |
+
|
| 11 |
+
- "models/yolox_l.onnx": the checkpoint for dwpose extraction.
|
| 12 |
+
|
| 13 |
+
- "models/v2-1_512-ema-pruned.ckpt": the checkpoint for Stable Diffusion.
|
| 14 |
+
|
| 15 |
+
|
| 16 |
+
|
| 17 |
+
|
| 18 |
+
## BibTeX
|
| 19 |
+
|
| 20 |
+
If this repo is useful to you, please cite our corresponding technical paper.
|
| 21 |
+
|
| 22 |
+
```bibtex
|
| 23 |
+
@article{wang2024unianimate,
|
| 24 |
+
title={UniAnimate: Taming Unified Video Diffusion Models for Consistent Human Image Animation},
|
| 25 |
+
author={Wang, Xiang and Zhang, Shiwei and Gao, Changxin and Wang, Jiayu and Zhou, Xiaoqiang and Zhang, Yingya and Yan, Luxin and Sang, Nong},
|
| 26 |
+
journal={arXiv preprint arXiv:2406.01188},
|
| 27 |
+
year={2024}
|
| 28 |
+
}
|
| 29 |
+
@inproceedings{TFT2V,
|
| 30 |
+
title={A Recipe for Scaling up Text-to-Video Generation with Text-free Videos},
|
| 31 |
+
author={Wang, Xiang and Zhang, Shiwei and Yuan, Hangjie and Qing, Zhiwu and Gong, Biao and Zhang, Yingya and Shen, Yujun and Gao, Changxin and Sang, Nong},
|
| 32 |
+
booktitle={CVPR},
|
| 33 |
+
year={2024}
|
| 34 |
+
}
|
| 35 |
+
@article{VideoComposer,
|
| 36 |
+
title={VideoComposer: Compositional Video Synthesis with Motion Controllability},
|
| 37 |
+
author={Wang, Xiang and Yuan, Hangjie and Zhang, Shiwei and Chen, Dayou and Wang, Jiuniu and Zhang, Yingya and Shen, Yujun and Zhao, Deli and Zhou, Jingren},
|
| 38 |
+
journal={NeurIPS},
|
| 39 |
+
year={2023}
|
| 40 |
+
}
|
| 41 |
+
```
|
uniAnimate/checkpoints/SodAnything_vits_24w_removeRelu_best_ckpt.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a92a82da6592d504ee909d41d6adb76b83bbca4d8c97c69afb7ecf7fd662231c
|
| 3 |
+
size 99538915
|
uniAnimate/checkpoints/arcface_w600k_r50.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4c06341c33c2ca1f86781dab0e829f88ad5b64be9fba56e56bc9ebdefc619e43
|
| 3 |
+
size 174383860
|
uniAnimate/checkpoints/big-lama.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7ba7aa7ac37a4d41fdbbeba3a2af7ead18058552997e3a3cd1a3b2210c9e6b4c
|
| 3 |
+
size 205803670
|
uniAnimate/checkpoints/configuration.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"framework":"Pytorch","task":"text-to-video-synthesis"}
|
uniAnimate/checkpoints/dw-ll_ucoco_384.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:724f4ff2439ed61afb86fb8a1951ec39c6220682803b4a8bd4f598cd913b1843
|
| 3 |
+
size 134399116
|
uniAnimate/checkpoints/iic/unianimate/.mdl
ADDED
|
Binary file (37 Bytes). View file
|
|
|
uniAnimate/checkpoints/iic/unianimate/.msc
ADDED
|
Binary file (555 Bytes). View file
|
|
|
uniAnimate/checkpoints/iic/unianimate/.mv
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
Revision:master,CreatedAt:1718280037
|
uniAnimate/checkpoints/inswapper_128.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e4a3f08c753cb72d04e10aa0f7dbe3deebbf39567d4ead6dce08e98aa49e16af
|
| 3 |
+
size 554253681
|
uniAnimate/checkpoints/models--LiheYoung--depth_anything_vits14/README.md
ADDED
|
@@ -0,0 +1,58 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
tags:
|
| 3 |
+
- depth_anything
|
| 4 |
+
- depth-estimation
|
| 5 |
+
---
|
| 6 |
+
|
| 7 |
+
# Depth Anything model, small
|
| 8 |
+
|
| 9 |
+
The model card for our paper [Depth Anything: Unleashing the Power of Large-Scale Unlabeled Data](https://arxiv.org/abs/2401.10891).
|
| 10 |
+
|
| 11 |
+
You may also try our [demo](https://huggingface.co/spaces/LiheYoung/Depth-Anything) and visit our [project page](https://depth-anything.github.io/).
|
| 12 |
+
|
| 13 |
+
## Installation
|
| 14 |
+
|
| 15 |
+
First, install the Depth Anything package:
|
| 16 |
+
```
|
| 17 |
+
git clone https://github.com/LiheYoung/Depth-Anything
|
| 18 |
+
cd Depth-Anything
|
| 19 |
+
pip install -r requirements.txt
|
| 20 |
+
```
|
| 21 |
+
|
| 22 |
+
## Usage
|
| 23 |
+
|
| 24 |
+
Here's how to run the model:
|
| 25 |
+
|
| 26 |
+
```python
|
| 27 |
+
import numpy as np
|
| 28 |
+
from PIL import Image
|
| 29 |
+
import cv2
|
| 30 |
+
import torch
|
| 31 |
+
|
| 32 |
+
from depth_anything.dpt import DepthAnything
|
| 33 |
+
from depth_anything.util.transform import Resize, NormalizeImage, PrepareForNet
|
| 34 |
+
from torchvision.transforms import Compose
|
| 35 |
+
|
| 36 |
+
model = DepthAnything.from_pretrained("LiheYoung/depth_anything_vitl14")
|
| 37 |
+
|
| 38 |
+
transform = Compose([
|
| 39 |
+
Resize(
|
| 40 |
+
width=518,
|
| 41 |
+
height=518,
|
| 42 |
+
resize_target=False,
|
| 43 |
+
keep_aspect_ratio=True,
|
| 44 |
+
ensure_multiple_of=14,
|
| 45 |
+
resize_method='lower_bound',
|
| 46 |
+
image_interpolation_method=cv2.INTER_CUBIC,
|
| 47 |
+
),
|
| 48 |
+
NormalizeImage(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
|
| 49 |
+
PrepareForNet(),
|
| 50 |
+
])
|
| 51 |
+
|
| 52 |
+
image = Image.open("...")
|
| 53 |
+
image = np.array(image) / 255.0
|
| 54 |
+
image = transform({'image': image})['image']
|
| 55 |
+
image = torch.from_numpy(image).unsqueeze(0)
|
| 56 |
+
|
| 57 |
+
depth = model(image)
|
| 58 |
+
```
|
uniAnimate/checkpoints/models--LiheYoung--depth_anything_vits14/config.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"encoder": "vits", "features": 64, "out_channels": [48, 96, 192, 384], "use_bn": false, "use_clstoken": false}
|
uniAnimate/checkpoints/models--LiheYoung--depth_anything_vits14/gitattributes
ADDED
|
@@ -0,0 +1,35 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
*.7z filter=lfs diff=lfs merge=lfs -text
|
| 2 |
+
*.arrow filter=lfs diff=lfs merge=lfs -text
|
| 3 |
+
*.bin filter=lfs diff=lfs merge=lfs -text
|
| 4 |
+
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
| 5 |
+
*.ckpt filter=lfs diff=lfs merge=lfs -text
|
| 6 |
+
*.ftz filter=lfs diff=lfs merge=lfs -text
|
| 7 |
+
*.gz filter=lfs diff=lfs merge=lfs -text
|
| 8 |
+
*.h5 filter=lfs diff=lfs merge=lfs -text
|
| 9 |
+
*.joblib filter=lfs diff=lfs merge=lfs -text
|
| 10 |
+
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
| 11 |
+
*.mlmodel filter=lfs diff=lfs merge=lfs -text
|
| 12 |
+
*.model filter=lfs diff=lfs merge=lfs -text
|
| 13 |
+
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
| 14 |
+
*.npy filter=lfs diff=lfs merge=lfs -text
|
| 15 |
+
*.npz filter=lfs diff=lfs merge=lfs -text
|
| 16 |
+
*.onnx filter=lfs diff=lfs merge=lfs -text
|
| 17 |
+
*.ot filter=lfs diff=lfs merge=lfs -text
|
| 18 |
+
*.parquet filter=lfs diff=lfs merge=lfs -text
|
| 19 |
+
*.pb filter=lfs diff=lfs merge=lfs -text
|
| 20 |
+
*.pickle filter=lfs diff=lfs merge=lfs -text
|
| 21 |
+
*.pkl filter=lfs diff=lfs merge=lfs -text
|
| 22 |
+
*.pt filter=lfs diff=lfs merge=lfs -text
|
| 23 |
+
*.pth filter=lfs diff=lfs merge=lfs -text
|
| 24 |
+
*.rar filter=lfs diff=lfs merge=lfs -text
|
| 25 |
+
*.safetensors filter=lfs diff=lfs merge=lfs -text
|
| 26 |
+
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
| 27 |
+
*.tar.* filter=lfs diff=lfs merge=lfs -text
|
| 28 |
+
*.tar filter=lfs diff=lfs merge=lfs -text
|
| 29 |
+
*.tflite filter=lfs diff=lfs merge=lfs -text
|
| 30 |
+
*.tgz filter=lfs diff=lfs merge=lfs -text
|
| 31 |
+
*.wasm filter=lfs diff=lfs merge=lfs -text
|
| 32 |
+
*.xz filter=lfs diff=lfs merge=lfs -text
|
| 33 |
+
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
+
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
+
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
uniAnimate/checkpoints/models--LiheYoung--depth_anything_vits14/pytorch_model.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ba0ab5365610467de4fea1ffe64b4ae3a756561a4df9e11bd7369205241f2082
|
| 3 |
+
size 99232841
|
uniAnimate/checkpoints/open_clip_pytorch_model.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9a78ef8e8c73fd0df621682e7a8e8eb36c6916cb3c16b291a082ecd52ab79cc4
|
| 3 |
+
size 3944692325
|
uniAnimate/checkpoints/portrate_selflabelFFinetune_fullAndCut_dp_size512_ImageANTIALIAS_adduserimage_best_ckpt.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c5be221ca1ac52d308634418abd430371a4473a0db5c7449cb63699ad4e8c675
|
| 3 |
+
size 110366705
|
uniAnimate/checkpoints/unianimate_16f_32f_non_ema_223000.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b78e68c02d5974d9f8198b2e96f6868c1b17921dea3eb7f729ce6dacdc481b45
|
| 3 |
+
size 5666897525
|
uniAnimate/checkpoints/v2-1_512-ema-pruned.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:88ecb782561455673c4b78d05093494b9c539fc6bfc08f3a9a4a0dd7b0b10f36
|
| 3 |
+
size 5214865159
|
uniAnimate/checkpoints/yoloface_8n.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:84d5bb985b0ea75fc851d7454483897b1494c71c211759b4fec3a22ac196d206
|
| 3 |
+
size 12639351
|
uniAnimate/checkpoints/yolov8m.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6c25b0b63b1a433843f06d821a9ac1deb8d5805f74f0f38772c7308c5adc55a5
|
| 3 |
+
size 52117635
|
uniAnimate/checkpoints/yolov8x.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c4d5a3f000d771762f03fc8b57ebd0aae324aeaefdd6e68492a9c4470f2d1e8b
|
| 3 |
+
size 136867539
|
uniAnimate/checkpoints/yolox_l.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7860ae79de6c89a3c1eb72ae9a2756c0ccfbe04b7791bb5880afabd97855a411
|
| 3 |
+
size 216746733
|