Testing accuracy of pretrained model (resnet18)

2 visualizaciones (últimos 30 días)
Nour Mohamed
Nour Mohamed el 25 de En. de 2021
Respondida: Gaurav Garg el 28 de En. de 2021
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.

Respuestas (1)

Gaurav Garg
Gaurav Garg el 28 de En. de 2021
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.

Categorías

Más información sobre Image Data Workflows en Help Center y File Exchange.

Community Treasure Hunt

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

Start Hunting!

Translated by