arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2404.09406cs.CVcs.HCcs.LGcs.RO
收藏

人机协同的多物种珊瑚图像分割

Human-in-the-Loop Segmentation of Multi-species Coral Imagery

QUT Centre for Robotics(QUT机器人中心);Image Analytics(图像分析);CSIRO Data61(CSIRO数据61)

Scarlett Raine, Ross Marchant, Brano Kusy, Frederic Maire, Niko Suenderhauf, Tobias Fischer

AI总结:

本文提出利用DINOv2特征和KNN实现高效珊瑚图像分割,仅用5个点标签即可在mIoU上提升13.5%。

摘要:

水下和水面机器人勘测产生的珊瑚礁图像量巨大,但领域专家对这些图像的标注成本高且耗时。点标签传播是一种利用已标注稀疏点图像生成增强真实数据的技术,可用于训练语义分割模型。本文表明,大基础模型的最新进展使仅使用去噪后的DINOv2基础模型提取的特征和K-最近邻(KNN)即可生成增强的真实掩码,无需预训练。对于极稀疏标签的图像,我们采用人机协同原则提升标注效率:若每张图像有5个点标签,我们的方法在mIoU上比现有最先进方法提升19.7%。当无法使用人机协同标注时,使用去噪DINOv2特征和KNN仍可使mIoU比现有最先进方法提升5.8%(5个网格点)。在语义分割任务中,仅使用5个点标签进行点标签传播时,我们比现有最先进方法在mIoU上提升13.5%。此外,我们还对点标签的数量和位置进行了全面研究,并提出了几种提高点标注图像效率的建议。

英文摘要:

Marine surveys by robotic underwater and surface vehicles result in substantial quantities of coral reef imagery, however labeling these images is expensive and time-consuming for domain experts. Point label propagation is a technique that uses existing images labeled with sparse points to create augmented ground truth data, which can be used to train a semantic segmentation model. In this work, we show that recent advances in large foundation models facilitate the creation of augmented ground truth masks using only features extracted by the denoised version of the DINOv2 foundation model and K-Nearest Neighbors (KNN), without any pre-training. For images with extremely sparse labels, we use human-in-the-loop principles to enhance annotation efficiency: if there are 5 point labels per image, our method outperforms the prior state-of-the-art by 19.7% for mIoU. When human-in-the-loop labeling is not available, using the denoised DINOv2 features with a KNN still improves on the prior state-of-the-art by 5.8% for mIoU (5 grid points). On the semantic segmentation task, we outperform the prior state-of-the-art by 13.5% for mIoU when only 5 point labels are used for point label propagation. Additionally, we perform a comprehensive study into the number and placement of point labels, and make several recommendations for improving the efficiency of labeling images with points.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑