This repo contains the code developed by me to perform multi-class segmentation on the architectural floor plan data contained in the Data folder.
Please install the modules given in the requirements.txt file.
-
Install packages via pip and requirements.txt
- pip install -r requirements.txt
-
Change the training hyperparameters if required in configuration.py file.
-
Run main.py file to initiate the training.
Please note that it is mandatory to have an account in Weights & Biases to monitor the progress of training.