Spaces:
Configuration error
Configuration error
Commit
·
61aa33b
1
Parent(s):
f2edec2
update annotator
Browse files- .gitignore +1 -0
- README.md +27 -1
.gitignore
CHANGED
@@ -1,4 +1,5 @@
|
|
1 |
annotator/ckpts/**
|
|
|
2 |
result/**
|
3 |
trash/**
|
4 |
data/**
|
|
|
1 |
annotator/ckpts/**
|
2 |
+
annotator/annotator_ckpts.tar.gz
|
3 |
result/**
|
4 |
trash/**
|
5 |
data/**
|
README.md
CHANGED
@@ -24,11 +24,37 @@ pip install -r requirements.txt
|
|
24 |
|
25 |
</details>
|
26 |
|
27 |
-
You may download all
|
28 |
```bash
|
|
|
29 |
bash download_all.sh
|
30 |
```
|
31 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
32 |
## 🔛 Prepare all the data
|
33 |
|
34 |
```
|
|
|
24 |
|
25 |
</details>
|
26 |
|
27 |
+
You may download all the base model checkpoints using the following bash command
|
28 |
```bash
|
29 |
+
## download sd 1.5, controlnet depth/pose v10/v11
|
30 |
bash download_all.sh
|
31 |
```
|
32 |
|
33 |
+
Prepare ControlNet annotator weights (e.g., DW-Pose, depth_zoe, depth_midas, OpenPose)
|
34 |
+
|
35 |
+
```bash
|
36 |
+
## Switch to the ckpts directory
|
37 |
+
cd ckpts
|
38 |
+
|
39 |
+
Method 1: Download individual models
|
40 |
+
Download the DW-Pose models (dw-ll_ucoco_384.onnx and yolo_l.onnx), as we found them to be more robust than OpenPose. (Note: Other models, such as depth_zoe, depth_midas, and OpenPose, can be automatically downloaded from HuggingFace.)
|
41 |
+
Available from:
|
42 |
+
- [Baidu](https://pan.baidu.com/s/1nuBjw-KKSxD_BkpmwXUJiw?pwd=28d7)
|
43 |
+
- [Google](https://drive.google.com/file/d/12L8E2oAgZy4VACGSK9RaZBZrfgx7VTA2/view?usp=sharing)
|
44 |
+
Download the detection model (yolox_l.onnx) from:
|
45 |
+
- [Baidu](https://pan.baidu.com/s/1fpfIVpv5ypo4c1bUlzkMYQ?pwd=mjdn)
|
46 |
+
- [Google](https://drive.google.com/file/d/1w9pXC8tT0p9ndMN-CArp1__b2GbzewWI/view?usp=sharing)
|
47 |
+
Then place both files into ./annotator/ckpts
|
48 |
+
|
49 |
+
Method 2: Download all annotator checkpoints in one package
|
50 |
+
# Note: This package includes all the required annotator models, such as DW-Pose, depth_zoe, depth_midas, and OpenPose. Requires approximately 4GB of storage space.
|
51 |
+
If you cannot access HuggingFace, you can download all the annotator checkpoints from:
|
52 |
+
- [BaiduYun](https://pan.baidu.com/s/1sgBFLFkdTCDTn4oqHjGb9A?pwd=pdm5)
|
53 |
+
- [Google](https://drive.google.com/file/d/1qOsmWshnFMMr8x1HteaTViTSQLh_4rle/view?usp=drive_link)
|
54 |
+
Then extract them into ./annotator/ckpts
|
55 |
+
|
56 |
+
```
|
57 |
+
|
58 |
## 🔛 Prepare all the data
|
59 |
|
60 |
```
|