基于注意力机制的TCS-YOLO船舶检测系统 |
| |
引用本文: | 龚思宇,陈姚节,陈黎.基于注意力机制的TCS-YOLO船舶检测系统[J].船舶工程,2023(11):108-115+144. |
| |
作者姓名: | 龚思宇 陈姚节 陈黎 |
| |
作者单位: | 武汉科技大学计算机科学与技术学院 |
| |
基金项目: | 国家自然科学基金项目(62271359); |
| |
摘 要: | 针对当前海上环境复杂、噪声干扰严重及船舶检测存在漏检和误检等问题,提出一种基于注意力机制的TCS-YOLO船舶检测系统。该算法使用K-Means++聚类算法确定目标样本的锚框以提高先验框与船舶目标的尺寸匹配度;在YOLOv7的Neck部分引入Transformer Block以捕获全局信息和丰富的上下文信息;在YOLOv7的Head部分添加CA注意力机制,有助于模型更准确地定位和识别感兴趣的对象。并在自制的船舶数据集上进行试验,结果表明,该算法的平均精度均值达到70.5%,相比原始的YOLOv7算法值提高了5.1%,能更准确地检测船舶,满足在复杂海上环境中进行船舶检测的需求。
|
关 键 词: | 船舶检测系统 注意力机制 YOLO |
|
|