首页 PaddleGAN 帖子详情
我想用PaddleGan训练一个条件GAN应该用哪种dataset呢?
收藏
快速回复
PaddleGAN 问答生成 503 2
我想用PaddleGan训练一个条件GAN应该用哪种dataset呢?
收藏
快速回复
PaddleGAN 问答生成 503 2

我想用我自己的数据集训练一个条件GAN。

看到官方给的配置文件中与条件GAN相关的只有configs/cond_dcgan_mnist.yaml,其中用到的数据集类型是CommonVisionDataset。但是这个数据集只能使用PaddlePaddle自带的数据集。如果我想用我自己的数据集(格式和MNIST一样,一张图片标注了一个类别),应该使用哪种dataset呢?

0
收藏
回复
全部评论(2)
时间顺序
FutureSI
#2 回复于2021-09

套件yaml文件里的数据集类型设置应该是用于一些公开数据集的预处理的,如果用自己的数据集,自己手动做那些transform和归一化,看看有没有UserDefined的之类的自定义类型。

0
回复
FutureSI
#3 回复于2021-09

详细的设置可以参考:https://github.com/PaddlePaddle/PaddleGAN

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户