Adjusting the Generative adversarial network example
显示 更早的评论
Hi. I want to feed grayscale images to the GAN and get grayscale images as output, but the generator layer is defined in a way that produces RGB images. How can I change this layer to produce grayscale images? Also, is there any other modifications needed in the training GAN provided by MATLAB in the following link?
I have tried changing the filter numbers of the last layers of both generator and discriminater from 3 to 1, but I end up with plain blue images during the learning process.
2 个评论
Eaby Babu
2021-1-14
Hi, Have you found a solution to train gray-scale images?
Natalia da Silva de Sá
2021-11-11
I fixed the blue image problem just changing the part of the code that display the images. So instead of image(I), I used imshow(I).
回答(1 个)
yanqi liu
2021-11-12
sir,may be use
augmentedImageDatastore(data,'ColorPreprocessing','gray2rgb');
类别
在 帮助中心 和 File Exchange 中查找有关 Deep Learning Toolbox 的更多信息
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!