稿件标题: | 面向无人机平台视觉任务的基于ViT的轻量型目标检测方法 |
稿件作者: | 高一博,杨传栋,陈栋 |
DOI: | 10.11809/bqzbgcxb2022.S2.042 |
栏目名称: | 信息科学与控制工程 |
关键词: | 无人机平台;目标检测;轻量化;自监督训练 |
文章摘要: | 设计了一种基于Vision Transformer结构的轻量化目标检测算法,前期采用一个小型CNN网络对图象做预处理提取初级特征,而后用多尺度特征金字塔ViT结构对不同分辨率特征图做注意力建模进行检测。算法采用窗口与全局注意力结合的方式减小推测时计算量预测加速。同时采用特征级掩码自监督训练方式对ViT结构做预训练。在实验中,所设计的两种不同层数模型均在保持实时检测速度的前提下达到了高于其他轻量算法的AP值。 |
引用本文格式: | 高一博,杨传栋,陈栋.面向无人机平台视觉任务的基于ViT的轻量型目标检测方法[J].兵器装备工程学报,2022,43(S2):241-246. GAO Yibo, YANG Chuandong, CHEN Dong.A ViTbased lightweight target detection method for UAV platform visual tasks[J].Journal of Ordnance Equipment Engineering,2022,43(S2):241-246. |
刊期名称: | 2022年增刊2期 |
出版时间: | 2023年1月 |
上线时间: | 2023年1月12日 |
浏览次数: | 1855 |
下载次数: | 31 |
免费阅读PDF 下载本期目录 下载本期封面 |