语义分割
用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
架构
ImageNet 预训练模型
下载 放在 models 目录内。
用法
数据预处理
该数据集包含SUNRGBD V1的10335个RGBD图像,执行下述命令提取训练图像:
$ python pre-process.py
像素分布:
数据集增强
图片
分割
图片
分割
训练
$ python train.py
如果想可视化训练过程,可执行:
$ t
2023-03-18 08:26:30
11.86MB
Python
1