This project provides a machine learning-powered FastAPI backend that determines the cleanliness of a room based on image input. The model can be used in automation tools to confirm whether a room is clean or not.
📦 Features
✅ Classifies images as clean (1) or dirty (0)
✅ Built with TensorFlow and FastAPI
✅ Easy-to-use API endpoint for predictions
✅ Ready for integration with web or mobile applications
🧠 How It Works
The model takes in an image input and returns a score between 0 and 1
A score closer to 1 means the room is clean, while a score closer to 0 means it’s dirty
For video streams, individual frames can be extracted, scored, and the average taken