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 个评论

Hi, Have you found a solution to train gray-scale images?
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 个)

sir,may be use
augmentedImageDatastore(data,'ColorPreprocessing','gray2rgb');

类别

帮助中心File Exchange 中查找有关 Deep Learning Toolbox 的更多信息

产品

版本

R2019b

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by