Change number of output classes in Squeezenet?

16 次查看（过去 30 天）

Asim Shahzad 2021-2-18

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/749282-change-number-of-output-classes-in-squeezenet

评论： Manav Madan 2022-6-22

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 个评论
显示 -1更早的评论隐藏 -1更早的评论

Manav Madan 2022-6-22

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.

请先登录，再进行评论。

请先登录，再回答此问题。