本仓库包含论文“用于RGB-D显着目标检测的数据级重组和轻量级融合方案”的代码实现。

代码结构

  • ./Ours 文件夹包含利用原始RGB和深度图计算显著性图的网络源代码。
  • ./Ours+ 文件夹包含改进网络的源代码,该网络用生成的显著性预测替换了原始深度图。

结果对比

| | Ours | Ours+ |

|---|---|---|

| | | |

评估指标

评估指标详见./Evaluation文件夹。

环境配置

  • 下载代码并解压缩至./目录下.
  • 下载预训练模型并存储至./model/目录下.
    • Ours.caffemodel 用于初始结果
    • Ours+.caffemodel 用于最终改进结果

运行测试

  1. ./Ours文件夹下运行test.m生成显著性图.
  2. ./Ours+文件夹下运行test+.m获得改进结果.

训练模型

  1. 下载训练数据集并解压缩至./Dataset/Train/目录下.
  2. 下载预训练模型并放置于./Model/目录下.
  3. 运行训练脚本 sh ./o