π§ Virtual Office Floor Planner (AI Edition)
An AI-powered application that allows users to replace floors in images using semantic segmentation. Built with Python, PyTorch, and Gradio.
- AI-Powered Floor Segmentation: Uses SegFormer model to accurately detect and segment floors in images
- Interactive Interface: Simple and intuitive Gradio-based web interface
- Real-time Processing: See results instantly
- Customizable: Easy to modify and extend for different use cases
- Python 3.8 or higher
- pip (Python package manager)
-
Clone the repository:
git clone https://github.com/krishnab0841/floor_replacement_app.git cd floor_replacement_app -
Create and activate a virtual environment (recommended):
python -m venv venv .\venv\Scripts\activate # On Windows # or source venv/bin/activate # On macOS/Linux
-
Install the required packages:
pip install -r requirements.txt
-
Start the application:
python App.py
-
Open your web browser and navigate to the URL shown in the terminal (usually
http://localhost:7860) -
Upload an office/room image and a carpet/flooring image to see the magic happen!
- Upload an office/room photo
- Upload a carpet/flooring pattern
- Click "Generate" to see the result
- gradio
- numpy
- Pillow
- torch
- transformers
This project is licensed under the MIT License - see the LICENSE file for details.
Contributions are welcome! Please feel free to submit a Pull Request.