A web application that allows the user to upload an image of a room and change the color of the walls or apply a selected texture.
I am using here two pretrained models. One for wall segmentation and another for wall position estimation.
This project utilizes code and a pretrained model for wall segmentation from the OriginalProject.
This project utilizes code and a pretrained model for Indoor Scene Layout Estimation from the OriginalProject.
Model weights was too large(600mb). You need to load it from drive and save at wall_estimation/weight/
.
Pre-trained weight at Google Drive.
-
Create a virtual environment:
python -m venv env
-
Start the virtual environment:
env\Scripts\Activate
-
Install dependencies:
pip install -r requirements.txt
-
Run:
python app.py
Folder test_images - contains images of rooms and textures to visualize the work