Skip to content

PriyankaNigade1992/CapstoneProject_DeepLearning

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 

Repository files navigation

CS668_CapstoneProject

Indoor scene recognition using Deep Learning Model

Dataset Used:

Dataset URL: https://bit.ly/41Fkh1m
Country: USA (MIT university)
15620 images in all 67 Indoor categories (Living Room, Airport)
A minimum of  100 photos in each category
Every image is in the jpg format.
The data set is 2.61 GB.

Screen Shot 2023-05-09 at 10 32 43 AM

Methodology:

Data Augmentation: Two transformation pipelines are utilized using PyTorch's transformations module to standardise input images.
The split-folders package is used to ensure an even distribution of each class across Train and test subsets.
Models utilized are: ResNet34, ResNext101_32x16d_wsl, GoogleNet and ShuffleNet.(Minimize risk of vanishing gradient, light weight, state-of-the-art performance , Requires limited computational resources)
Model selection is based on complexity, accuracy, and suitability for the available computational resources and dataset size.
Model accuracy is used as a measure of performance.

Testing

Screen Shot 2023-05-09 at 10 37 26 AM
Screen Shot 2023-05-09 at 10 37 11 AM

About

Indoor scene recognition using Deep Learning Model

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published