Commit ·
275e335
1
Parent(s): 6772976
Use images in 'images' folder in example scripts
Browse files
README.md
CHANGED
|
@@ -39,7 +39,7 @@ from diffusers.utils import load_image
|
|
| 39 |
from easy_dwpose import DWposeDetector
|
| 40 |
|
| 41 |
|
| 42 |
-
pose_image = load_image("./pose_image_1.png")
|
| 43 |
|
| 44 |
# Load detector
|
| 45 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
|
@@ -77,6 +77,9 @@ image = pipe(
|
|
| 77 |
image=skeleton,
|
| 78 |
generator=torch.manual_seed(97),
|
| 79 |
).images[0]
|
|
|
|
|
|
|
|
|
|
| 80 |
```
|
| 81 |
|
| 82 |
Generated pose is:
|
|
@@ -103,7 +106,7 @@ from diffusers.utils import load_image
|
|
| 103 |
from easy_dwpose import DWposeDetector
|
| 104 |
|
| 105 |
|
| 106 |
-
pose_image = load_image("./pose_image_2.png")
|
| 107 |
|
| 108 |
# Load detector
|
| 109 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
|
@@ -143,6 +146,9 @@ image = pipe(
|
|
| 143 |
image=skeleton,
|
| 144 |
generator=torch.manual_seed(79),
|
| 145 |
).images[0]
|
|
|
|
|
|
|
|
|
|
| 146 |
```
|
| 147 |
|
| 148 |
Generated pose is:
|
|
|
|
| 39 |
from easy_dwpose import DWposeDetector
|
| 40 |
|
| 41 |
|
| 42 |
+
pose_image = load_image("./images/pose_image_1.png")
|
| 43 |
|
| 44 |
# Load detector
|
| 45 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
|
|
|
| 77 |
image=skeleton,
|
| 78 |
generator=torch.manual_seed(97),
|
| 79 |
).images[0]
|
| 80 |
+
|
| 81 |
+
skeleton.save("./images/dwpose_1.png")
|
| 82 |
+
image.save("./images/dwpose_image_1.png")
|
| 83 |
```
|
| 84 |
|
| 85 |
Generated pose is:
|
|
|
|
| 106 |
from easy_dwpose import DWposeDetector
|
| 107 |
|
| 108 |
|
| 109 |
+
pose_image = load_image("./images/pose_image_2.png")
|
| 110 |
|
| 111 |
# Load detector
|
| 112 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
|
|
|
| 146 |
image=skeleton,
|
| 147 |
generator=torch.manual_seed(79),
|
| 148 |
).images[0]
|
| 149 |
+
|
| 150 |
+
skeleton.save("./images/dwpose_2.png")
|
| 151 |
+
image.save("./images/dwpose_image_2.png")
|
| 152 |
```
|
| 153 |
|
| 154 |
Generated pose is:
|