-
公开(公告)号:CN118172736A
公开(公告)日:2024-06-11
申请号:CN202410173774.8
申请日:2024-02-07
Applicant: 北京交通大学
IPC: G06V20/54 , G06V10/764 , G06V10/82 , G06N3/09
Abstract: 本发明公开了一种用于智能交通的实时检测方法及装置,该方法包括:获取实时采集的道路或航拍图像,输入预先建立和训练好的检测模型,得到待检测目标的位置大小及类别;所述检测模型包括骨干层、FPN层、三个结构相同的检测头,及NMS网络;其中,所述骨干层和所述FPN层,用于对采集的图像进行下采样,提取特征,得到类图和特征图;所述检测头,用于将类图和特征图通过生成类图网络转化为稀疏特征,再经过稀疏卷积后分别输入检测框回归网络和分类网络,得到待检测目标的位置大小及类别;所述NMS网络,用于对三个检测头分别得到的待检测目标的位置大小及类别,进行非极大值抑制,得到待检测目标的最终位置大小及类别。
-
公开(公告)号:CN118172736B
公开(公告)日:2025-01-24
申请号:CN202410173774.8
申请日:2024-02-07
Applicant: 北京交通大学
IPC: G06V20/54 , G06V10/764 , G06V10/82 , G06N3/09
Abstract: 本发明公开了一种用于智能交通的实时检测方法及装置,该方法包括:获取实时采集的道路或航拍图像,输入预先建立和训练好的检测模型,得到待检测目标的位置大小及类别;所述检测模型包括骨干层、FPN层、三个结构相同的检测头,及NMS网络;其中,所述骨干层和所述FPN层,用于对采集的图像进行下采样,提取特征,得到类图和特征图;所述检测头,用于将类图和特征图通过生成类图网络转化为稀疏特征,再经过稀疏卷积后分别输入检测框回归网络和分类网络,得到待检测目标的位置大小及类别;所述NMS网络,用于对三个检测头分别得到的待检测目标的位置大小及类别,进行非极大值抑制,得到待检测目标的最终位置大小及类别。
-