naver-ai / StyleMapGAN

Official pytorch implementation of StyleMapGAN (CVPR 2021)
https://www.youtube.com/watch?v=qCapNyRA_Ng
Other
458 stars 81 forks source link

How to do StyleMixing on custom dataset #25

Open ammar-deep opened 2 years ago

ammar-deep commented 2 years ago

Hello. Thanks for this code. I have trained the model on my own dataset that is different from the face dataset. The image reconstruction, random generation works well. However, If I do the style mixing their is a hard coded pkl file required that is related to the celeba_hq dataset (data/celeba_hq/local_editing/celeba_hq_test_GT_sorted_pair.pkl).

Just for the sake of running stylemixing I downloaded the dataset and ran the code, however, the results are not good.

Could you please share your thoughts on how to apply stylemixing on a custom dataset?