公开(公告)号：CN114241053B

公开(公告)日：2024-05-28

申请号：CN202111662790.6

申请日：2021-12-31

Applicant: 北京工业大学

Inventor： 杨志伟 ,  同磊 ,  段娟 ,  肖创柏

IPC: G06T7/73 ,  G06V20/40 ,  G06V20/52 ,  G06V10/774 ,  G06V10/80 ,  G06V10/82 ,  G06V10/98 ,  G06V10/44 ,  G06N3/045 ,  G06N3/0464

Abstract: 本发明公开了基于改进的注意力机制FairMOT多类别跟踪方法，本发明充分利用无人机拍摄的视频，通过引入通道和空间注意力机制的多目标多类别目标追踪算法，对无人机拍摄的视频序列中存在的目标进行追踪及其运动轨迹可视化显示。具体包括如下步骤，首先对无人机数据集进行预处理；构建模型；训练网络；多类别多目标追踪算法的性能评估；在FairMOT的DL34‑base的基础上添加了注意力机制，使网络更好的学习图像中的语义信息和空间信息。还对网络的其它结构进行了修改，在目标检测分支中修改heatmap检测头与box size检测头，使得目标检测分支对于目标中心点位置和目标尺寸的预测的精确度更高，进而对整个模型的追踪性能有了不错的提升，本发明具有更高的追踪精准度。

公开(公告)号：CN114241053A

公开(公告)日：2022-03-25

申请号：CN202111662790.6

申请日：2021-12-31

Applicant: 北京工业大学

Inventor： 杨志伟 ,  同磊 ,  段娟 ,  肖创柏

IPC: G06T7/73 ,  G06V20/40 ,  G06V20/52 ,  G06V10/774 ,  G06V10/80 ,  G06V10/82 ,  G06V10/98 ,  G06K9/62 ,  G06N3/04 ,  G06N3/08

Abstract: 本发明公开了基于改进的注意力机制FairMOT多类别跟踪方法，本发明充分利用无人机拍摄的视频，通过引入通道和空间注意力机制的多目标多类别目标追踪算法，对无人机拍摄的视频序列中存在的目标进行追踪及其运动轨迹可视化显示。具体包括如下步骤，首先对无人机数据集进行预处理；构建模型；训练网络；多类别多目标追踪算法的性能评估；在FairMOT的DL34‑base的基础上添加了注意力机制，使网络更好的学习图像中的语义信息和空间信息。还对网络的其它结构进行了修改，在目标检测分支中修改heatmap检测头与box size检测头，使得目标检测分支对于目标中心点位置和目标尺寸的预测的精确度更高，进而对整个模型的追踪性能有了不错的提升，本发明具有更高的追踪精准度。