Универсальный алгоритм сегментации на основе динамического балансирования и усиления признаков

DENG Wanyu ,  

LIU Yangwu ,  

摘要

Хотя визуальные признаки заметных объектов и замаскированных объектов противоположны, оба требуют высокой точности сегментации границ. Однако существующие методы часто сталкиваются с ограничениями производительности при обобщении между задачами из-за предпочтения однородных признаков. В связи с этим в данной работе предложена универсальная модель сегментации SAM2-UFFM. Модель использует иерархический каркас SAM2 в качестве кодировщика, замораживая предварительно обученные параметры и вводя легковесный адаптер для эффективной тонкой настройки параметров, полностью сохраняя мощные возможности представления признаков; отказалась от традиционной архитектуры U-Net и разработала декодер AttUNet, который через полномасштабные скип-соединения обеспечивает эффективное объединение информации на всех уровнях и решает проблему потери информации; предложен модуль внимания сбалансированных признаков, который в сочетании с модулем доработки взаимодополняющих признаков динамически балансирует глобальную семантику и локальные детали, затем целенаправленно усиливает глобальное восприятие заметных объектов и точное сегментирование границ замаскированных объектов, сопровождаемое стратегией глубокого контроля для предотвращения пере-сегментации. Эксперименты показали, что модель превосходит современные методы на трех публичных наборах данных. Структурные метрики на наборах COD и SOD улучшились на 1,02% и 1,83% соответственно по сравнению с лучшими существующими моделями, а средняя абсолютная ошибка снизилась до 0,035 и 0,018. Кроме того, в задаче сегментации дефектов поверхности стали IoU повысился на 2,77%, что эффективно подтверждает способность модели к междоменной генерализации.

关键词

сегментация изображения; заметный объект; замаскированный объект; SAM2; механизм внимания

阅读全文

以上内容由讯飞翻译自动生成,翻译内容仅供参考。对于因使用本网站翻译内容产生的相关后果,本网站不承担任何商业和法律责任。

The above content is generated by Large Model Translation. The translated content is for reference only. We do not assume any commercial or legal responsibilty for any consequences arising from the use of our website