Room clasification using pytorch and flask (it is capable of classifying rooms but can be extened to detect object in each room)
-
before running any file first download weights : https://drive.google.com/open?id=1-2NykomtvREWVWrhJAaj0XMh4lArytAK
-
install all requirements:
> pip install -r requirements.txt
- run a app.py file:
> python app.py