dimitribarbot commited on
Commit
275e335
1 Parent(s): 6772976

Use images in 'images' folder in example scripts

Browse files
Files changed (1) hide show
  1. README.md +8 -2
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: