- Install packages.
pip install -r requirements.txt
- Download r3d dataset to
dataset/r3d.tfrecords
and convert it to csv file.
python readTFRecord.py
python img2csv.py
- Train the network,
python main.py
- Deploy the network,
python deploy.py
Simple replication of 'Deep FloorPlan Recognition using a Multi-task Network with Room-boundary-Guided Attention' using Pytorch.
Python