vipulheybuddy/roomtesting

Room Wall Visualizer

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

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.

Getting Started

1. Create a virtual environment:

   python -m venv env
   

2. Start the virtual environment:

   env\Scripts\Activate
   

3. Install dependencies:

   pip install -r requirements.txt
   

4. Run:

   python app.py
   

Open http://localhost:9000

Folder test_images - contains images of rooms and textures to visualize the work