Testing accuracy of pretrained model (resnet18)

조회 수: 13(최근 30일)
Nour Mohamed
Nour Mohamed 2021년 1월 25일
답변: Gaurav Garg 2021년 1월 28일
I tested the pretrained resnet18 network available on MATLAB using CIFAR-100 dataset (100 class dataset) and the accuracy is very low, even though resnet18 is an object classification network.
How can I get the actual accuracy? because I saw in some papers using the same CIFAR dataset they got around 70% accuracy.

답변(1개)

Gaurav Garg
Gaurav Garg 2021년 1월 28일
Hi Nour,
I hope you are using transfer learning to train the network 'resnet-18'. You can look at example here for more guidance on transfer learning, though the given example is for AlexNet.
If the model still doesn't give good accuracy, it can be because of many reasons, a few of which have been listed here:
  1. Imbalanced Data classes - It could happen that a few classes have many data points and other classes have a few points. To remove this ambiguity for the model, kindly follow the link.
  2. Data Augmentation - You can augment the images (link). This helps to increase the accuracy of the model by giving a few more training points.

Community Treasure Hunt

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

Start Hunting!

Translated by