How to add new classes into a pre-trained network?

Question

Da Huang 2020-8-9

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/577066-how-to-add-new-classes-into-a-pre-trained-network

回答： Farshid PirahanSiah 2020-10-19

Say I already have a pre-trained network who can classify class A, class B and class C. Is there anyway I can re-train it only using some data from a new class D (instead of using a full dataset that contains data from all four classes), so that the new network is capable of classify both class A, B, C and D? Also wondering if it's achieveable by changing some certain settings during the transfer learning process?

Thanks in advance.

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

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

Answer 1

Gouri Chennuru 2020-8-13

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/577066-how-to-add-new-classes-into-a-pre-trained-network#answer_479382

Hi Da Huang,

You can load the model, train and remove the last layer, change its other neurons and train the network again! This will basically train the the last layer from the beginning.

or, you can also specify the number of classes in the last fully connected layer of your network as the Output size argument.

You can refer to this link for more information.

Hope this Helps!