Oct, 2023
使用SAM进行建筑物分割模型的零样本精细调整
Zero-Shot Refinement of Buildings' Segmentation Models using SAM
TL;DR我们提出了一种新颖的方法来针对现有模型的泛化性能下降问题,将基础模型自适应于特定领域,如遥感图像。我们通过集成预训练的卷积神经网络作为提示生成器,为Segment Anything Model增加了识别能力,从而改善了其在遥感图像上的表现。通过在三个遥感数据集上评估我们的方法,包括WHU Buildings数据集、Massachusetts Buildings数据集和AICrowd Mapping Challenge数据集,我们观察到我们的方法在不同分布的性能上取得了显著提高,并计划发布我们的代码库,以促进遥感领域对基础模型进行更多领域特定任务的探索。