Семантическая сегментация изображений с малым числом образцов на основе трансформера с двойным перекрёстным вниманием

LIU Yu ,  

GUO Yingchun ,  

ZHU Ye ,  

YU Ming ,  

摘要

Семантическая сегментация изображений с малым числом образцов позволяет сегментировать новые классы, используя лишь небольшое количество примеров. Для решения проблемы недостаточного извлечения семантической информации в существующих методах в данной статье предлагается метод семантической сегментации изображений с малым числом образцов на основе сети с двойным перекрёстным вниманием. Метод использует структуру Transformer, применяя модуль двойного перекрёстного внимания для одновременного изучения дальних зависимостей между многоуровневыми признаками запросов и опорными признаками по канальному и пространственному измерениям. Во-первых, в статье представлен модуль перекрёстного внимания по каналам, который в сочетании с модулем перекрёстного позиционного внимания образует модуль двойного перекрёстного внимания. Модуль перекрёстного внимания по каналам изучает семантические взаимосвязи по каналам между признаками запроса и опоры, а модуль перекрёстного позиционного внимания захватывает удалённые контекстные зависимости между признаками запроса и опоры. Затем несколько модулей двойного перекрёстного внимания обеспечивают многоуровневые интерактивные признаки с богатой семантической информацией для изображения запроса. В конце вводится вспомогательная функция потерь, и посредством методов апсемплинга и остаточных связей многоуровневые интерактивные признаки подключаются к декодеру для получения точных результатов сегментации новых классов. Метод достиг mIoU 69,9% (1-shot) и 72,4% (5-shot) на наборе данных PASCAL-5i, и 48,9% (1-shot) и 54,6% (5-shot) на наборе данных COCO-20i. По сравнению с основными методами, производительность сегментации данного подхода достигла передового уровня.

关键词

Семантическая сегментация изображений с малым числом образцов;Структура Transformer;Перекрёстное внимание по каналам;Двойное перекрёстное внимание;Вспомогательные потери

阅读全文

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

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