Skip to content

sunt40/Semantic-Segmentation

 
 

Repository files navigation

语义分割

用SegNet进行室内语义分割。

依赖

数据集

image

按照 说明 下载 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

架构

image

ImageNet 预训练模型

下载 VGG16 放在 models 目录内。

用法

数据预处理

该数据集包含SUNRGBD V1的10335个RGBD图像,执行下述命令提取训练图像:

$ python pre-process.py

像素分布:

image

数据集增强

图片 分割 图片 分割
image image image image
image image image image
image image image image
image image image image
image image image image

训练

$ python train.py

如果想可视化训练过程,可执行:

$ tensorboard --logdir path_to_current_dir/logs

image

演示

下载 预训练模型 放在 models 目录,然后执行:

$ python demo.py

图例:

image

输入 真实 输出
image image image
image image image
image image image
image image image
image image image
image image image
image image image
image image image
image image image
image image image

About

室内语义分割

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 100.0%