摘要
本发明公开了一种基于热传导原理的U-ViT网络小目标检测方法,其实现步骤是,构建并训练热传导启发的U-ViT小目标检测网络,利用该网络中的热传导启发的注意力机制单元传导和聚合图片不同方向的特征图,利用该网络中的热传导残差单元对特征图进行二阶差分操作以精确提取小目标的边缘信息,利用该网络中的热传导启发的transformer子模块关注小目标局部细节的同时捕捉像素点的长期依赖关系。本发明主要解决对噪声干扰严重、对比度低的图像中的弱小目标进行检测的问题,具有边界划分明确容易提取小目标的完整特征,特征图的提取兼顾细节与整体,避免了小目标的检测结果的不完全预测的问题,稳定性、可信性强,降低漏检率与虚警率的优点。
- 单位