Comments

You must log in or register to comment.

nutpeabutter t1_iuvkakt wrote

Try dreambooth. 600 is far too little for training a stylegan from scratch (heavy augmentation could help but I doubt it)

5

TheDigitalRhino t1_iuw2bum wrote

It maybe worth it to use Stable Diffusion. Since the data set is smaller, it may not be realistic to expect you to create thousands of more.

3

Current-Basket3920 t1_iuxezmx wrote

The code is freely available and on Youtube you find videos on „How to use Stylegan“.

But:

  • As already mentioned you‘ll need more data. 5-10k are already at the low end. You need more like 50-100k I think.
  • You need some serious hardware to train something like this. For Stylegan they used 8 high end GPUs for a week. I guess for Stylegan2/3 even more. You might be happy with less than NVIDIA, but it‘s no quick thing to do.
  • And even then there‘s no guarantee that it will turn out perfectly. It might need some fine-tuning, it‘s not a simple algorithm.
2

thePsychonautDad t1_iuyz5ob wrote

As others have said, you don't have enough examples to train a stylegan, but you can retrain Stable Diffusion on that few examples. That'll allow you to render anything you can imagine in your style too for example.

1