Specify Output Class For GAN Image Generator
8 views (last 30 days)
Show older comments
james flynn on 17 Apr 2021
Commented: Mahesh Taparia on 15 Sep 2021
In MATLAB's 'Train Generative Adversarial Network' example, the final step is to generate new images using the predict function and the following code.
%% Generate New Images
numObservations = 25;
ZNew = randn(numLatentInputs,numObservations,'single');
dlZNew = dlarray(ZNew,'CB');
if (executionEnvironment == "auto" && canUseGPU) || executionEnvironment == "gpu"
dlZNew = gpuArray(dlZNew);
dlXGeneratedNew = predict(dlnetGenerator,dlZNew);
I = imtile(extractdata(dlXGeneratedNew));
I = rescale(I);
The example produces a random sample of 25 images from 5 classes. I want to modify this to request one image from a specific class. For example, generate 5 images for the 'Dandilion' class. Can someone help me understand if it's possible to request specific generator outputs like this? Thanks in advance.
Mahesh Taparia on 21 Apr 2021
You can possibly try with conditional GAN to generate images of a particular class. For more information, you can refer this documentation of conditional GAN. You can refer here for image generation.
Mahesh Taparia on 15 Sep 2021
It is already answered in the community. You can refer the below links:
More Answers (0)
Find more on Image Data Workflows in Help Center and File Exchange
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!Start Hunting!