Skip to content

Fine Grained Image Classification with Class Imbalance using Bilinear EfficientNet with Focal Loss and Label Smoothing

Notifications You must be signed in to change notification settings

jimitshah77/plant-pathology

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

19 Commits
 
 
 
 
 
 

Repository files navigation

Bilinear EfficientNet with Focal Loss + Label Smoothing

Plant Pathology

python

Problem Objectives:

The aim of this challenge is to build a Generalised Model for the task of Image Classification. We have to also deal with Class Imbalance Problem and detect Fine Grained details which differentiates whether the leaf is diseased or Not.

Solution:

i) Image Classification: EfficientNet

ii) Class Imbalance and Generalization : Focal Loss + Label Smoothing

iii) Detect Fine Grain Details: Bilinear CNN

Model: BiLinear EfficientNet with Focal Loss+ Label Smoothing

python

Expriments and Results:

a) Selecting the Best and Efficient CNN Backbone (img size:380x380):
python

b) Exprimenting with Loss Functions (img size:380x380):
python

c) Bilinear CNN (img size:760x760):

python

python

References:

(1) EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks (http://https//arxiv.org/abs/1905.11946)
(2) Focal Loss for Dense Object Detection (https://arxiv.org/abs/1708.02002)
(3) Bilinear CNNs for Fine-grained Visual Recognition (https://arxiv.org/abs/1504.07889)
(4) Focal Loss Implementation (https://github.com/aldi-dimara/keras-focal-loss)
(5) Label Smoothing Explaination and Formulas (https://towardsdatascience.com/what-is-label-smoothing-108debd7ef06)
(6) BiLinear CNN Implementation (https://medium.com/@scorrea92/bilinear-cnn-models-in-tensorflow-keras-801121cc8c4d)

Notebook: https://www.kaggle.com/jimitshah777/bilinear-efficientnet-focal-loss-label-smoothing
Checkout my rank in this Kaggle Competition here: https://www.kaggle.com/jimitshah777

About

Fine Grained Image Classification with Class Imbalance using Bilinear EfficientNet with Focal Loss and Label Smoothing

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published