Skip to content

149189/Transfer_Learning_on_10_Food_Classes

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits
 
 
 
 
 
 

Repository files navigation

Transfer_Learning_on_10_Food_Classes

I trained a model to classify images into 10 different food categories using transfer learning. I utilized two powerful pre-trained models as feature extractors:

ResNet 50 V2

image

EfficientNetB0.

image

Key Steps:

  1. Data Preprocessing: The dataset consisted of 10 food classes, and images were resized and normalized to match the input requirements of the models.

  2. Transfer Learning:

    1. ResNet 50 V2 was used as a feature extractor, leveraging its deep residual network architecture to capture complex features from the images.
    
    2. EfficientNetB0 was also employed for its efficiency in balancing performance and computational cost, particularly with its scalable architecture.
    
  3. Model Training: The extracted features from both models were fed into fully connected layers for classification.

  4. Evaluation: Both models were evaluated on the same test dataset to compare their performance.

image

About

Applied Transfer Learning on 10 Food Classes

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published