-
Notifications
You must be signed in to change notification settings - Fork 14
Open
Description
Hi,
I can see that you need to load pose images "_pose.jpg" to train stage 2, but I can't find them in the prepared dataset you provided.
PCDMs/src/dataset/stage2_dataset.py
Line 99 in 8f6fc81
| s_pose = Image.open(s_img_path.replace("/train_all_png/", "/openpose_all_img/").replace(".png", "_pose.jpg")).convert("RGB").resize(self.size, Image.BICUBIC) |
Would you like to upload them to Google Drive later or do we have to generate them from the normalized_pose_txt ourselves?
Extra question: Why are we using pose "images" in stage 2 instead of coordinates as we did in stage 1? I understand that this is borrowed from ControlNet, but I would like to know the motivation behind it.
Thanks
Metadata
Metadata
Assignees
Labels
No labels