Deep Floor Plan Recognition using a Multi-task Network with Room-boundary-Guided Attention代码实现

代码地址：https://github.com/zlzeng/DeepFloorplan

关于python2.7转python3.5.

环境，tensorflow  1.15.0

           cuda 10.0 + cudnn 7.6

测试代码：

下载与训练模型并放置：./pretrained/下

存在的问题：

1. AttributeError: 'dict' object has no attribute 'iteritems'

 

解决：

将iteritems改为items

训练代码：

1. python3 中取消了 range 函数，而把 xrange 函数重命名为 range，所以现在直接用 range 函数即可。

但是xrange的类型是List，range的类型是range对象，所以要将xrange()修改为list(range())。

2. 将下载的数据文件放入./dataset 中