Легковесный высокоточный алгоритм обнаружения объектов на основе фреймворка YOLO

FAN Xin-chuan ,  

CHEN Chun-mei ,  

摘要

В условиях взаимного ограничения точности обнаружения и системных затрат в алгоритмах обнаружения объектов с мультиразмерами изображения предложен легковесный высокоточный алгоритм обнаружения объектов на основе фреймворка YOLO. На базе сети MobileNetv3 в рамках YOLO были усовершенствованы понижающая выборка и механизм внимания по каналам для точного извлечения признаков объектов, что снизило ненужные затраты. Спроектирована структура пирамиды признаков с одностадийным безголовым слиянием для построения разных рецептивных полей с целью получения информации разных масштабов, что усилило адаптивность алгоритма к объектам разных размеров. Одновременно был применён SIOU в качестве регрессионной функции потерь и Soft-NMS для обработки избыточных рамок, что повысило точность алгоритма. Экспериментальные результаты на наборах данных MS COCO и UA-DETRAC показали, что предложенный в работе улучшенный алгоритм по сравнению с исходным YOLOXs при сохранении точности уменьшил количество параметров модели и вычислительные затраты на 64,98% и 57,14% соответственно; на датасете UA-DETRAC достижение mAP@0.5 составило 70,5%, что выше на 3,52%, при этом FPS увеличилось на 14,4%. Улучшенный алгоритм значительно снизил системные затраты, повысил точность и эффективно обеспечил двойную производительность обнаружения.

关键词

Обнаружение объектов;YOLOXs;Механизм внимания;Легковесный;SIOU;Soft-NMS

阅读全文

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

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