Training NCSNPP on a custom 512x512 dataset

[ysig]

Hi!

Firstly thanks a lot for your amazing work and contribution!

I was really motivated to try your code in a real custom dataset of about 140K pictures in 512x512.
Following I would like to use NCSNPP, but I am having a really hard time understanding how I can load my own data.

Could you maybe describe programmatically what is the pipeline of doing so?
i.e.

• what script should I run to create a dataset and how and
• which config file should I edit and which lines/attributes should I change - at least to have a basic starting point?

Thanks in advance!
ysig

[Jelly954]

Hi!
Is your problem solved? I also want to try to train the model using my own data set. Do you know how to load your own data set? Does the training set only contain clear images?
Thanks in advance!
Jelly

[qianduodu]

Is your problem solved? I also want to try to train the model using my own data set. Do you know how to load your own data set? Does the training set only contain clear images?