Feature extraction using CNN and classification with SVM

5 次查看（过去 30 天）
Huseyin Nasifoglu
Huseyin Nasifoglu 2021-3-11
回答： Raynier Suresh 2021-3-17
Hi,
I have a question on feature extraction from 2D CNN and classifying features with SVM. First let me introduce what I am trying to do;
1) I use pretrained network AlexNet which is trained with ImageNet.
2) I have a small dataset and use transfer learning for the classification problem. First, I trained my database with AlexNet by retraining all the parameters inside the network (no freezing layers) and observed the accuracy.
3) Now I want to classify the extracted features from the network with SVM. Should I use the initial AlexNet network's layer for feature extraction (default AlexNet) or the retrained network's layer on step 2?
I actually tried both of them and acquired higher accuracy on retrained network with almost %20 difference compared with the initial AlexNet. Is it just because the retrained parameters performed well on SVM? That's why a higher accuracy observed? Which one should be used?

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

采纳的回答

Raynier Suresh
Raynier Suresh 2021-3-17
Hi, It usually depends on the problem and the approach would change for different problems, there is no single answer for it. Based on the data and the problem you are trying to solve choose the most suitable method.

更多回答（0 个）

类别

在 Help Center 和 File Exchange 中查找有关 Image Data Workflows 的更多信息

标签

Community Treasure Hunt

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

Start Hunting!

Translated by