用GAN训练自己的数据生成新的图片
出于工作需要,学习了GAN,原理这块就不多讲了,主要讲怎么训练自己的数据生成新的图片,因为博客上大多是生成MNIST数据集,生成自己的图片时,有些小坑。下面记录一下本人基于参考链接,将MNIST数据集的代码改成生成自己数据时遇到的坑。一、读取数据问题# MNIST datasetmnist = datasets.MNIST( root='./data/', train=True, transform=img_transform, download=True)# Data loaderda
原创
2021-08-02 15:25:33 ·
13028 阅读 ·
52 评论