一种目标检测模型构建方法、目标检测方法及相关装置
Abstract:
本发明属于图像多目标检测技术领域,公开了一种目标检测模型构建方法、目标检测方法及相关装置;其中,所述目标检测模型构建方法包括:以基于YOLO框架的目标检测网络为基准模型,将基准模型的特征提取网络中的所有卷积层替换为重排网络模块和设置于重排网络模块后的无参数注意力模块,获得轻量化目标检测网络;基于选定的训练数据集对所述轻量化目标检测网络进行深度学习预训练,然后基于教师模型使用置信度蒸馏损失进行蒸馏训练,达到预设收敛条件后,构建获得目标检测模型。本发明构建获得的目标检测网络在面临复杂的检测场景和实时的检测任务时,具有较高的检测精度和检测效率。
Patent Agency Ranking
0/0