IDEA-Research / HumanSD

[ICCV 2023] The official implementation of paper "HumanSD: A Native Skeleton-Guided Diffusion Model for Human Image Generation"
Apache License 2.0
271 stars 18 forks source link

How to generate poses for custom images. Please guide. #25

Closed hacker009-sudo closed 10 months ago

hacker009-sudo commented 10 months ago

@juxuan27 , Can you please explain how to create .npz files for custom dataset.

hacker009-sudo commented 10 months ago

@juxuan27 , Actually I need to know how we can prepare our custom dataset. Please guide.

juxuan27 commented 10 months ago

Hi, to prepare your custom dataset, you need to prepare image-pose annotation-text annotation pairs. Specifically, image path and text annotation can be directly putted in mapping_file.json as our given LAION-Human. As for the pose annotation, we generate it directly with MMPose and provide the npz file path in mapping_file.json.

xiao2mo commented 6 months ago

please give a toy demo for generate such kind of feature using related ckpt.

xiao2mo commented 6 months ago

@juxuan27