Метод YOLO-DSM для обнаружения мелких дефектов с множеством фонов на панелях TFT-LCD

KONG Xiangfei ,  

WANG Sen ,  

ZHAO Lin ,  

CHEN Mingfang ,  

摘要

Предложена модель глубокого обучения для обнаружения изображений на основе YOLO-DSM. Во-первых, после каждого модуля Dark введён модуль HMU для повышения точности обнаружения дефектов на TFT-LCD панелях. Оригинальный SPP заменён на SSMA, что позволило сети больше концентрироваться на объектах с низкой контрастностью на фоне. Во-вторых, введён модуль DSM для помощи сети в усилении полезных признаков и подавлении бесполезных, а также для улучшения интеграции семантической информации. Наконец, исходная сверточная операция с понижением размерности заменена модулем ODConv для уточнения локальных карт признаков и полного извлечения локальных признаков дефектов. На самодельном наборе данных дефектов TFT-LCD проведено сравнение с современными алгоритмами. Результаты показывают, что сеть YOLO-DSM достигла точности mAP в 97,40% и FPS 77,42, что соответствует требованиям к обнаружению дефектов TFT-LCD.

关键词

Мелкие визуальные дефекты;YOLO-DSM;Полноразмерная динамическая свёртка;Механизм внимания SCSE

阅读全文