Effacer les filtres
Effacer les filtres

Change number of output classes in Squeezenet?

16 vues (au cours des 30 derniers jours)
Asim Shahzad
Asim Shahzad le 18 Fév 2021
Commenté : Manav Madan le 22 Juin 2022
I want to retrain a slightly modified squeezenet on the CIFAR100 dataset. Here are the last few layers of Squeezenet:
Usually, a fully connected layer is replaced to change the number of output classes, or the pooling layer is changed. However, MATLAB's Deep Network Designer doesn't give any options to adjust output size of the globalAveragePooling2dLayer. And Squeezenet doesn't have any fully connected layers.
My question is, how do I change the network output class size from 1000 (imagenet) to 100 (cifar100)?
  1 commentaire
Manav Madan
Manav Madan le 22 Juin 2022
First store the values set for different features in last 4 layer the "convolution2dLayer" then delete and select a new "convolution2dLayer" where copy rest of the values for parameters and change "NumFilters" to the number of classes you have.

Connectez-vous pour commenter.

Réponses (0)

Catégories

En savoir plus sur Image Data Workflows dans Help Center et File Exchange

Produits


Version

R2020b

Community Treasure Hunt

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

Start Hunting!

Translated by