Uneven color distribution in stable and artistic model outputs about deoldify HOT 1 CLOSED

As far as datasets that would help in terms of good results across all categories of images- I'd say Google's Open Images dataset ( https://storage.googleapis.com/openimages/web/index.html ) is probably your best bet. It's huge in comparison to what DeOldify was originally trained on, and is much more diverse. But the problems go beyond just data. I've moved on past this particular training approach years ago, but if you were to improve upon it, my guess is increasing the batch size would go a long way. I had very limited compute resources when training this (1080 TI gpu with 11 GB VRAM!). A lot of things have changed in the past 5 years!

Since it has been 5 years, there's definitely better approaches out there, too. I've seen diffusion based stuff and I'd say that's probably the way to go now ( https://palette.fm ).

from deoldify.