How do you get the Classification Layer Probabilities for pretrained CNNs ?
7 vues (au cours des 30 derniers jours)
Afficher commentaires plus anciens
Hello,
I have code that runs through a file of images and classifies each, using the pretrained Squeezenet network. I want to get the results of the classification layer and save all the probabilities and the end classification choice. How do I do this?
Thank you!
for k = 1:numFiles
try
full_path=fullfile(imgs(k).folder, imgs(k).name);
mydata{k} = imread(full_path);
i = imresize(mydata{k},[227,227]);
a = classify(net, i)
imshow(i); title(char(a))
catch ME
fprintf("The imread call threw the following error:\n%s\n", ME.message)
fprintf("The error identifier for this error is:\n%s\n", ME.identifier)
end
end
0 commentaires
Réponse acceptée
Divya Gaddipati
le 9 Avr 2021
You can use activations to get the output at a certain layer
out = activations(net,X,'layerName');
For more information, refer to the link below:
Plus de réponses (0)
Voir également
Catégories
En savoir plus sur Deep Learning Toolbox dans Help Center et File Exchange
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!