XiangpengYang commited on
Commit
61aa33b
·
1 Parent(s): f2edec2

update annotator

Browse files
Files changed (2) hide show
  1. .gitignore +1 -0
  2. 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 data and checkpoints using the following bash command
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
  ```