Clean Vs Messy Rooms Classifier using Tensorflow

Overview

In this project, the task is to classify the image of a room as clean or messy. The model architecture is built using the Convolutional Neural Network (CNN) from the Tensorflow library.

Dataset

The data set for this project was obtained from Kaggle. Pictures consist of two classes, clean and messy rooms. There are 192 images in the training set (96 per class); 20 images (10 per class) in the validation set; 10 images (5 per class) in the test set. The location is diverse, including bedrooms, living rooms, dining rooms, study rooms and kitchens, which might help to discourage the model from capturing unrelated features but instead focus more on the "messiness".

Work Steps

1. Import library
2. Download and extract file
3. Separate training and validation set
4. Data pre-processing using image augmentation
5. Prepare train data
6. Building a model architecture with CNN
7. Create Callbacks
8. Plotting accuracy and loss
9. Predict image

Model Summary

Visualize Accuracy and Loss

Predict Model