Skip to content

In this project I will finetune pretrained VGG16 model using transfer learning to detect clean or messy room

Notifications You must be signed in to change notification settings

Utshav-paudel/Messy-or-CleanRoom-Detection

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Messy-or-CleanRoom-Detection

In this project I had finetuned pretrained VGG16 model using transfer learning to detect clean or messy room.

Reasons for using Transfer Learning in this project :

Due to the limitation of training data transfer learning was best option for this project.

  • Only had 190 images for training
  • Only had 20 images for validation
  • Only had 10 images for test

1. Transferring pretrained model and finetuning it

2. Preparing training , test set with some preprocessing and fitting the model

3. Making prediction

Predictions made

About

In this project I will finetune pretrained VGG16 model using transfer learning to detect clean or messy room

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published