基于SAM的水陆两栖环境感知微调策略与应用

SAM-Based Fine-Tuning Strategy and Application of Amphibious Environment Perception

  • 摘要: 针对水陆两栖无人平台在不确定环境中面临的高误报率及多感知任务整合困难的问题,本研究提出了一种基于分割一切模型(segment anything model, SAM)的多模型联合环境感知方法,实现了障碍物检测与水陆域分割的统一处理. 具体而言,是将U-Net和YOLOv8与SAM结合,U-Net和YOLOv8负责获取目标的粗略轮廓,而SAM通过其编码−解码结构实现进一步精细分割. 此外,设计了专门的微调策略以实现联合训练,进一步提升了模型的性能. 本研究还构建了专有数据集USV-Dataset,并开发了数据引擎以提高标注效率. 为增强模型的泛化能力,采用了4个公开数据集与USV-Dataset进行混合训练,涵盖了多样化的场景和障碍物类别. 实验结果表明,该方法实现了96.8%的mPA分割精度和10 FPS的推理速度,展现出良好的泛化能力,能够满足中低速两栖无人平台的实时环境感知需求.

     

    Abstract: In view of the high false alarm rate and multi-sensory task integration challenges faced by amphibious unmanned platforms in uncertain environments, in this study a multi-model joint environment perception method based on the segment anything model (SAM) was proposed, which achieved unified processing of obstacle detection and amphibious domain segmentation. Specifically, U-Net and YOLOv8 were combined with SAM. U-Net and YOLOv8 were responsible for obtaining the rough outline of the target, while SAM achieved fine segmentation through its encoding-decoding structure. In addition, a special fine-tuning strategy was designed to achieve joint training, which further improved the performance of the model. In this study, a proprietary dataset USV-Dataset was also constructed and a data engine was developed to improve the annotation efficiency. In order to enhance the generalization ability of the model, four public datasets were used for mixed training with USV-Dataset, covering a variety of scenarios and obstacle categories. Experimental results show that this method achieves 96.8% mPA segmentation accuracy and 10 FPS inference speed, showing good generalization ability and meeting the real-time environment perception needs of medium and low-speed amphibious unmanned platforms.

     

/

返回文章
返回
Baidu
map