/ImageClassification

Classify household images into different categories

Primary LanguageJupyter Notebook

ImageClassification

Classify household images into different categories.

Used ResNet-50 as the training model (pretrained model from PyTorch).

Dataset is the household images in 5 shots. Training accuracy achieved 100% and testing accuracy is 81%.