dimitribarbot
commited on
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:
|