利索能及
我要发布
收藏
专利号: 202110633467X
申请人: 重庆大学
专利类型:发明专利
专利状态:已下证
更新日期:2025-12-22
缴费截止日期: 暂无
联系人

摘要:

权利要求书:

1.一种夜间车辆检测方法,其特征在于,该方法包括:获取待检测的图像;

利用训练好的目标检测模型对所述待检测的图像进行检测,得到检测目标;其中,所述目标检测模型对待检测的图像的处理过程包括:对待检测的图像进行特征提取,得到图像特征;

对所述图像特征进行特征增强,得到增强特征;

将所述增强特征输入到RPN网络中,生成候选框;

通过ROIPooling层对所述候选框进行处理,得到固定大小的特征图;

对所述特征图进行回归和分类,得到检测目标。

2.根据权利要求1所述的夜间车辆检测方法,其特征在于,所述对待检测的图像进行特征提取,得到图像特征的步骤包括:通过ResNet50网络对所述待检测的图像进行特征提取;其中,在ResNet50的Conv3至Conv5阶段的残差模块中引入缩放点乘注意力。

3.根据权利要求1所述的夜间车辆检测方法,其特征在于,所述对所述图像特征进行特征增强,得到增强特征的步骤包括:特征金字塔结构FPN对所述图像特征进行特征增强,得到增强特征。

4.根据权利要求3所述的夜间车辆检测方法,其特征在于,所述特征金字塔结构FPN包括四个特征层P2,P3,P4,P5,对特征层P2、P3进行下采样操作,对特征层P5进行上采样。

5.根据权利要求1所述的夜间车辆检测方法,其特征在于,所述对所述特征图进行回归和分类,得到检测目标的步骤包括:将所述固定大小的特征图输入到两个并列的全连接层中,对特征图进行回归和分类,得到检测目标。

6.根据权利要求5所述的夜间车辆检测方法,其特征在于,使用Softmax分类器对特征图进行分类,其中,Softmax分类器的公式为:其中,Softmaxj是Softmax分类器输出向量的第j个值,aj为输入向量的第j个值,ak为输入向量的第k个值,T表示类别数。

7.根据权利要求6所述的夜间车辆检测方法,其特征在于,若存在多个重叠的候选框,采用Soft‑NMS方法对候选框进行优化,Soft‑NMS的公式为:其中,Nt为设定的阈值,M为置信度最高的候选框,bi为其他任一候选框,IoU()表示交并比。

8.一种夜间车辆检测装置,其特征在于,包括:图像获取模块,用于获取待检测的图像;

目标检测模块,用于利用训练好的目标检测模型对所述待检测的图像进行检测,得到检测目标;其中,所述目标检测模型包括:特征提取模块,用于对待检测的图像进行特征提取,得到图像特征;

特征增强模块,用于对所述图像特征进行特征增强,得到增强特征;

候选框生成模块,用于将所述增强特征输入到RPN网络中,生成候选框;

特征图生生成模块,用于通过ROI Pooling层对所述候选框进行处理,得到固定大小的特征图;

分类回归模块,用于对所述特征图进行回归和分类,得到检测目标。