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 中

你可能感兴趣的:(图像分割-边缘检测)