search

ChrisChen1023 / HINT

HINT: High-quality INpainting Transformer with Enhanced Attention and Mask-aware Encoding
MIT License
30 stars 4 forks source link

Dataset preparation #9

Closed Rahaf-jaber closed 5 months ago

Rahaf-jaber commented 7 months ago

Hi I didn't know which file I should download if I want to train it using the CelebA-HQ dataset, and how the dataset folders should look to work well with code?

Thanks.

ChrisChen1023 commented 7 months ago

Hi I didn't know which file I should download if I want to train it using the CelebA-HQ dataset, and how the dataset folders should look to work well with code?

Thanks.

Hi,

For training on CelebA-HQ, you need to download the officially publiced CelebA-HQ dataset. Then, use the /script/flist.py file to generate the .flst file and set the path of the .flist file in your config fiile. Finally, runthe train.py to train the model. Hope this is helpful.

Cheers.