search

zoubohao / DenoisingDiffusionProbabilityModel-ddpm-

This may be the simplest implement of DDPM. You can directly run Main.py to train the UNet on CIFAR-10 dataset and see the amazing process of denoising.
MIT License
1.48k stars 156 forks source link

条件生成中自定义数据集标签制作 #13

Open yangshenchang opened 1 year ago

yangshenchang commented 1 year ago

大佬，请教一个问题，我正在使用你的算法生成车牌数据，在随机生成下效果还不错，现在要使用条件生活，为固定的车牌号生成对应车牌，但是这个工程中label条件只是单独的0-9的数字，请问一下如果要使用7-8位的车牌数据做条件进行生成，需要改哪些地方

ONobody commented 1 year ago

@yangshenchang 你好，请问这个怎样训练自己的数据集 数据种类只有8类的时候需要改那些谢谢

yangshenchang commented 1 year ago

@ONobody 看你自己的数据集，如果只是分类数据的话，直接用torchvision.datasets.ImageFolder函数读取就可以，8个类别在对应8个文件夹应该就能训练；如果你的类别比较复杂，比如像我的车牌一样，那就需要自定义dataset函数，将你的类别编码，然后对应修改modelcondition.py文件中cond_embedding的维度就可以了

ONobody commented 1 year ago

blade 你好 我用1700多张图片训练了200个eopch 但生成的效果是这样的 是我有那个地方不对吗

ONobody commented 1 year ago

@yangshenchang