metadata
license: mit
tags:
- image-classification
- room-classification
- pytorch
- computer-vision
datasets:
- custom-room-dataset
pipeline_tag: image-classification
widget:
- src: https://example.com/bedroom.jpg
example_title: Bedroom Example
- src: https://example.com/kitchen.jpg
example_title: Kitchen Example
Room Classification Model
Model Details
- Model Info: A lightweight classifier that determines if the indoor property image is a Bathroom, Bedroom, Dining Room, Kitchen or Living Room
- Model Type: ResNet50-based room classifier
- Classes: 5 room types (bath, bed, din, kitchen, living)
- Accuracy: 87% on validation set
- Input Size: 224x224 RGB images
Usage
- Provided in example_usage.py file