使用带权重图与动态用户交互的 U-Net 进行交互式分割
计算机视觉与模式识别
2021-11-19 v1
摘要
交互式分割近期因需专家输入以进一步提升分割性能的专业任务而受到关注。本工作中，我们提出一种新颖的交互式分割框架，其中用户点击根据当前分割掩码动态调整尺寸。被点击区域构成权重图，并作为新颖的加权损失函数输入深度神经网络。为评估我们的损失函数，采用了一种交互式 U-Net（IU-Net）模型，该模型将前景与背景用户点击作为交互的主要方式。我们在 BCV 数据集上训练与验证，同时在来自 MSD 数据集的脾脏与结肠癌 CT 图像上测试，以相比标准 U-Net 利用我们的加权损失函数提升整体分割精度。应用动态用户点击尺寸仅通过单次用户交互即分别将整体精度提升了 5.60% 与 10.39%。
引用
@article{arxiv.2111.09740,
title = {Interactive segmentation using U-Net with weight map and dynamic user interactions},
author = {Ragavie Pirabaharan and Naimul Khan},
journal= {arXiv preprint arXiv:2111.09740},
year = {2021}
}