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
Downloads last month
5
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support