- 专利标题: 基于频域自注意力机制的夜间目标检测、训练方法及装置
-
申请号: CN202210902801.1申请日: 2022-07-29
-
公开(公告)号: CN114972976A公开(公告)日: 2022-08-30
- 发明人: 章依依 , 虞舒敏 , 应志文 , 郑影 , 徐晓刚 , 王军
- 申请人: 之江实验室
- 申请人地址: 浙江省杭州市余杭区中泰街道之江实验室南湖总部
- 专利权人: 之江实验室
- 当前专利权人: 之江实验室
- 当前专利权人地址: 浙江省杭州市余杭区中泰街道之江实验室南湖总部
- 代理机构: 杭州浙科专利事务所
- 代理商 孙孟辉; 杨小凡
- 主分类号: G06V20/00
- IPC分类号: G06V20/00 ; G06V10/48 ; G06V10/56 ; G06V10/60 ; G06V10/764 ; G06V10/77 ; G06V10/82
摘要:
本发明公开了基于频域自注意力机制的夜间目标检测、训练方法及装置,采用具有夜间图像检测标签的开源数据集训练检测网络,首先将输入图像进行预处理,使其从RGB转化为YUV通道,分别将三个通道的图像划分成多个区块;对于三个通道的各个区块,分别进行DCT离散余弦变换;将各个区块中属于同一频域的信息依据其原本的空间关系存入同一通道中,生成数个代表不同频域的通道;将所有频域通道输入自注意力网络模块,该模块通过计算各通道之间的可缩放点积自注意力,输出每个通道动态加权后的数值,再将其分别输入相同的多层感知器MLP;将输出结果输入检测网络中,最终获得图像的检测结果。
公开/授权文献
- CN114972976B 基于频域自注意力机制的夜间目标检测、训练方法及装置 公开/授权日:2022-12-20