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%.
Classify household images into different categories