This is our repository for the final project of CSE 4622 MACHINE LEARNING LAB, KIRT.
KIRT is a real time mobile application which classifies common household objects. We experimented with 3 seperate base architectures VGG-16, ResNet50, MobileNetV2.
Link to dataset is here
Model | Accuracy | Precision | Recall | F1 Score |
---|---|---|---|---|
VGG-16 | 50.36% | 55.76% | 44.34% | 39.85% |
ResNet50 | 70.05% | 72.19% | 64.48% | 64.46% |
MobileNetV2 | 23.33% | 27.21% | 19.87% | 16.01 |
Download Trained VGG-16 model here
Download Trained ResNet50 model here
Download Trained MobileNetV2 model here
Download App Ready ResNet50 model here