stylegan2-tf-2.x icon indicating copy to clipboard operation
stylegan2-tf-2.x copied to clipboard

Published 20 hours ago • verified publisher icon moono

Please update readme

Open siavash-khodadadeh opened this issue 4 years ago • 3 comments

Hello,

This is a very useful repo. Can you please update readme so one can figure out how to train on custom datasets?

Thank you, Siavash

siavash-khodadadeh avatar Sep 25 '20 15:09 siavash-khodadadeh

To use custom dataset, one needs to replace dataset_ffhq.py file which returns batch of image within value range of [-1, 1] with size of [None, 3, res, res] where res is your desired image size.

And I think handling tensorflow's data is out of the scope of this repository. There are so many different input types to cover, so this link will help you how to load data as in tf.data.Dataset.

moono avatar Sep 28 '20 00:09 moono

Awesome, thanks.

siavash-khodadadeh avatar Sep 28 '20 01:09 siavash-khodadadeh

Hi, how would you integrate labels into this? Is there support for multiple labels?

hdubbs avatar Oct 15 '21 11:10 hdubbs