用SegNet进行室内语义分割。
按照 说明 下载 SUN RGB-D 数据集,放在 data 目录内。
$ wget http://3dvision.princeton.edu/projects/2015/SUNrgbd/data/SUNRGBD.zip
$ wget http://3dvision.princeton.edu/projects/2015/SUNrgbd/data/SUNRGBDtoolbox.zip
下载 VGG16 放在 models 目录内。
该数据集包含SUNRGBD V1的10335个RGBD图像,执行下述命令提取训练图像:
$ python pre-process.py
像素分布:
图片 | 分割 | 图片 | 分割 |
---|---|---|---|
$ python train.py
如果想可视化训练过程,可执行:
$ tensorboard --logdir path_to_current_dir/logs
下载 预训练模型 放在 models 目录,然后执行:
$ python demo.py
图例:
输入 | 真实 | 输出 |
---|---|---|