Use GAN discrimator for single-class classification
4 Ansichten (letzte 30 Tage)
Ältere Kommentare anzeigen
Hello,
i try to implement a neural network for classifying different defects for quality inspection. I want to use a single-class classification.
To accomplish this, i want to train a generative adversarial networks and use the discriminator for classification.
So i used the sunflower-example for implementing my first GAN.
In this example, there is a line which "classifies" the generated outputs with the help of the discriminator network:
dlYPredGenerated = forward(dlnetDiscriminator, dlXGenerated);
I expected the output to consist of 2 labels: "Original" or "Fake". Instead, i get a long list of numbers:
(:,:,1,1) =
5.9427
(:,:,1,2) =
7.5930
(:,:,1,3) =
9.3393
etc.
How can i use the resulting discriminator after training a GAN with a set of images? I just want to have a 2 - label classification, consisting of "original" and "fake".
0 Kommentare
Antworten (0)
Siehe auch
Kategorien
Mehr zu Get Started with Deep Learning Toolbox finden Sie in Help Center und File Exchange
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!