/PyTorch-DeepFloorplan

Simple replication of 'Deep FloorPlan Recognition using a Multi-task Network with Room-boundary-Guided Attention' using Pytorch.

Primary LanguagePython

PyTorch DeepFloorplan

How to run?

  1. Install packages.
pip install -r requirements.txt
  1. Download r3d dataset to dataset/r3d.tfrecords and convert it to csv file.
python readTFRecord.py
python img2csv.py
  1. Train the network,
python main.py
  1. Deploy the network,
python deploy.py