Skip to content

chandra-ps612/plant_diseases_classification_model

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 

Repository files navigation

plant_diseases_classification_model

For this particular project, I used TensorFlow==2.6.4, EfficientNetB0 pre-trained model for Transfer_learning freezing the base_model. Regarding Fine-tune_model, I used a separate code block to explain the process. The model predicts label_name by providing image_path from test_set.

What is EfficientNet?

It was introduced by GoogleAI in 2019. It is a CNN architecture and scaling method that uniformly scales all dimensions of CNN such as depth, width, and resolution using a compound co-efficient.

It provides a way to scale up CNNs in a more structured manner while also balancing all dimensions of the network at once, leading to a significant improvement in both accuracy and efficiency.

EfficientNet has 8 models from b0 to b7. Each model has 4 components given below-

  1. Stem layer
  2. Final Layer
  3. Sub-blocks (Each block has sub-blocks.)
  4. Modules (Each sub-blocks has modules.)

Layer details:

Input Layer -> Rescaling -> Normalization -> Zero Padding -> Conv2D -> Batch Normalization -> Activation

Models' details:

Base Model Resolution (Input shapes)
EfficientNetB0 224
EfficientNetB1 240
EfficientNetB2 260
EfficientNetB3 300
EfficientNetB4 380
EfficientNetB5 456
EfficientNetB6 528
EfficientNetB7 600

Reference:

  1. Dataset link: https://lnkd.in/eVyxTqrX