Skip to content

Compare CAM results for cat2dog images for each model of EfficientNet-B0, EfficientNet-B1, VGG16, and ResNet50. Through this, we select the most efficient way to give Feature Importance.

Notifications You must be signed in to change notification settings

kyugorithm/Efficientnet_CAM_comparison

Repository files navigation

First of all, I inform you that I made the code using the following.
The performance of EfficientNet should be similar, but VGG or ResNet appeared to be superior.
However, in fact, the accuracy verified by someone is more than 98% based on B0. If you find a problem, please suggest an issue.

: https://www.kaggle.com/bassbone/dog-vs-cat-transfer-learning-by-pytorch

CAM comparison among 4 famous models (EfficientNet-B0, EfficientNet-B1, VGG16, and ResNet50) on Cat2Dog dataset

Compare CAM results for cat2dog images for each model of EfficientNet-B0, EfficientNet-B1, VGG16, and ResNet50.
Through this, we select the most efficient way to give Feature Importance.
image
image

link : https://www.kaggle.com/c/dogs-vs-cats

EfficientNet

Indirect performance verification of EfficientNet is possible below.
This is the classification performance obtained using the vast amount of data from ImageNet.
image

Quantitative model performance comparison

Since the total amount of learning is 5 epoch, the results cannot be considered with perfect accuracy.
Furthermore the resolution is set to 128 and there is much room for accuracy to increase if the resolution is increased.
VGG and ResNet used the Pre-trained model, while EfficientNet was trained from the beginning.

update)
Accuracy

  1. EN:B0(96.1%)
  2. EN:B1(96.6%)
  3. VGG16(96.9%)
  4. RES50(98.0%)
    interence time (per sample)
  5. EN:B0(x)
  6. EN:B1(6.8ms)
  7. VGG16(8.4ms)
  8. RES50(8.1ms)
    image
    image
    image

About

Compare CAM results for cat2dog images for each model of EfficientNet-B0, EfficientNet-B1, VGG16, and ResNet50. Through this, we select the most efficient way to give Feature Importance.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages