Cathrineee / ImageClassification

Classify household images into different categories

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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%.

About

Classify household images into different categories


Languages

Language:Jupyter Notebook 100.0%