我想用PaddleGan训练一个条件GAN应该用哪种dataset呢?
收藏
我想用我自己的数据集训练一个条件GAN。
看到官方给的配置文件中与条件GAN相关的只有configs/cond_dcgan_mnist.yaml,其中用到的数据集类型是CommonVisionDataset。但是这个数据集只能使用PaddlePaddle自带的数据集。如果我想用我自己的数据集(格式和MNIST一样,一张图片标注了一个类别),应该使用哪种dataset呢?
0
收藏
请登录后评论
套件yaml文件里的数据集类型设置应该是用于一些公开数据集的预处理的,如果用自己的数据集,自己手动做那些transform和归一化,看看有没有UserDefined的之类的自定义类型。
详细的设置可以参考:https://github.com/PaddlePaddle/PaddleGAN