MATLAB Answers

Specify Output Class For GAN Image Generator

2 ビュー (過去 30 日間)
james flynn
james flynn 2021 年 4 月 17 日
コメント済み: Mahesh Taparia 2021 年 9 月 15 日
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);
end
dlXGeneratedNew = predict(dlnetGenerator,dlZNew);
I = imtile(extractdata(dlXGeneratedNew));
I = rescale(I);
figure
image(I)
axis off
title("Generated Images")
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
Mahesh Taparia 2021 年 4 月 21 日
Hi
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.
  2 件のコメント

サインインしてコメントする。

その他の回答 (0 件)

Community Treasure Hunt

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

Start Hunting!

Translated by