How can I add additional features to a pretrained AlexNet?

1 vue (au cours des 30 derniers jours)
I am classifying images using transfer learning and the pretrained AlexNet convolutional neural network.
I would like to add additional features in the fully connected layers at the end of the network.
Is it possible to add features that bypass the convolutional layers and are incorporated only at the fully connected layers?

Réponse acceptée

MathWorks Support Team
MathWorks Support Team le 26 Août 2021
Modifié(e) : MathWorks Support Team le 26 Août 2021
You can do this by following the workflow:
1. Extract the features that is outputted by the pretrained  (& transfer learning) AlexNet network right before the fully connected layer:
2. Concatenate the features from step 1 with the additional features from your image metadata.
3. Create and train a fully-connected neural network that will take in the concatenated features from step 2.

Plus de réponses (0)

Catégories

En savoir plus sur Deep Learning Toolbox dans Help Center et File Exchange

Tags

Community Treasure Hunt

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

Start Hunting!

Translated by