Skip to content

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

License

Notifications You must be signed in to change notification settings

zcemycl/PyTorch-DeepFloorplan

Repository files navigation

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

About

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

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages