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 follow the *.parquet files to download the images #11

Closed ucasyjz closed 1 year ago

juxuan27 commented 1 year ago

Hi, @yang326922943 ! You may refer to https://github.com/rom1504/img2dataset/blob/main/dataset_examples/laion-aesthetic.md to download images with .parquet files. Noted that the order of Laion-Human is different with that in the link. You may need to make some changes to the code.

ucasyjz commented 1 year ago

Thank you! Moreover, we find that the images do not match the poses, can you give me a guidance?

juxuan27 commented 1 year ago

Our images do not use initial .parquet files in Laion, but use the .parquet files provided in our google drive. You may check the correct .parquet files and it will match the pose annotations in our google drive.

juxuan27 commented 1 year ago

You can check https://github.com/IDEA-Research/HumanSD/issues/4 for details.

zhhezhhe commented 1 year ago

Will you provide a download script?

juxuan27 commented 1 year ago

Hi, @zhhezhhe . We have provided the download script, please refer to the readme.