alexzhou907 / DDBM

138 stars 14 forks source link

Unpaired AFHQ dataset #6

Open douhongkun opened 4 months ago

douhongkun commented 4 months ago

I have tried using this repo on the AFHQ dataset (unpaired animal faces) and it did require changing the load_data() function, yeah. You will also need to create a new dataloader for your own dataset.

I trained the model to convert dogs to cats. However, when I ran inference from the following dog images, I got the corresponding cat images:

image

As you can see, there aren't really strong similarities between the input and output images like you would expect in something like CycleGAN. It's possible I just didn't train the model long enough, or the architecture was too small, or I used the wrong parameters. Either way, I haven't been able to get it to work for unpaired data, but I would be interested to hear if anyone has been able to get this to work, especially the authors.

Originally posted by @harveymannering in https://github.com/alexzhou907/DDBM/issues/3#issuecomment-2104686846

IqraNosheen786 commented 3 months ago

Hello @douhongkun , I'm also trying to run this code on my own dataset but getting NaN values, Can you guide me on how you adjusted this code for another dataset? WhatsApp Image 2024-06-13 at 17 27 58_0c82640f

dang123454 commented 1 month ago

Hello, @IqraNosheen786 , I have the same problem, Have you solved it yet?

chasecjg commented 6 days ago

Hello @douhongkun , I'm also trying to run this code on my own dataset but getting NaN values, Can you guide me on how you adjusted this code for another dataset? WhatsApp Image 2024-06-13 at 17 27 58_0c82640f

I have the same problem. Have you solved your problem?