Skip to content

Latest commit

 

History

History
10 lines (4 loc) · 318 Bytes

README.md

File metadata and controls

10 lines (4 loc) · 318 Bytes

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