发明

一种基于知识蒸馏的跨模态图像旋转目标识别方法及系统2024

2024-10-16 12:23:57 发布于四川 12
  • 申请专利号:CN202411259450.2
  • 公开(公告)日:2024-12-27
  • 公开(公告)号:CN118781331A
  • 申请人:湖南大学
摘要:本发明公开了一种基于知识蒸馏的跨模态图像旋转目标识别方法及系统,本发明包括构建和训练以红外模态图像与可见光模态图像作为输入的教师网络模型,以可见光模态图像/红外模态图像作为输入的学生网络模型,利用可见光模态图像的样本训练学生网络模型,将可见光模态图像及其对应的红外模态图像的样本输入教师网络模型,利用教师网络模型中的多模态主干网、多尺度聚合层中提取的特征进行知识蒸馏并计算特征和关系蒸馏损失来训练优化学生网络模型的网络参数,直至完成对学生网络模型的训练以用于对输入的可见光模态图像进行旋转目标识别。本发明旨提高学生网络模型的检测精度,同时兼具跨模态检测功能。

专利内容

本发明公开了一种基于知识蒸馏的跨模态图像旋转目标识别方法及系统,本发明包括构建和训练以红外模态图像与可见光模态图像作为输入的教师网络模型,以可见光模态图像/红外模态图像作为输入的学生网络模型,利用可见光模态图像的样本训练学生网络模型,将可见光模态图像及其对应的红外模态图像的样本输入教师网络模型,利用教师网络模型中的多模态主干网、多尺度聚合层中提取的特征进行知识蒸馏并计算特征和关系蒸馏损失来训练优化学生网络模型的网络参数,直至完成对学生网络模型的训练以用于对输入的可见光模态图像进行旋转目标识别。本发明旨提高学生网络模型的检测精度,同时兼具跨模态检测功能。G06V10/25(2022.01);G06N3/084(2023.01);G06N3/096(2023.01);G06V10/40(2022.01);G06V10/774(2022.01);G06V10/778(2022.01);G06V10/80(2022.01);G06V10/82(2022.01)

最新专利