Giters

yuanming-hu / exposure

Learning infinite-resolution image processing with GAN and RL from unpaired image datasets, using a differentiable photo editing model.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How can I train the model using my own data?

cchen156 opened this issue · comments

commented

For example, I have two datasets of PNG images. How can I use the code to train the model, which will transfer the style of one dataset to the other one (unpaired)?

Thank you!

commented

Hi,

Sorry for the long delay. I get absorbed into several other projects.

Please do a "git pull" and "git submodule update".
You can train using images in folders using the new FolderDataProvider.
Take a look at the end of config_sintel.py as an example.
To train using the Sintel dataset, do "python3 train.py sintel test"

Hope it's still helpful. Btw, I could not come up a good example without copyright issue to demonstrate this usage. If you know any dataset that I can use as a demo (without copyright issues) for such user provided inputs, please let me know...
I think professional photos can be good inputs but the copyright issues are making me crazy...

Best,
Yuanming

commented

Hi Yuanming,

Thank you for your update! For dataset, you are free to use our data: https://github.com/cchen156/Learning-to-See-in-the-Dark

Thank you!

commented

Thanks! They are interesting data and I will have a try when possible.