目标检测模型的训练方法、装置、电子设备及介质
Abstract:
本发明实施例公开了一种目标检测模型的训练方法、装置、电子设备及介质,其中,该方法包括:获取多组训练样本数据,其中,所述训练样本数据包括样本图像以及所述样本图像对应的目标锚框信息;基于所述目标锚框信息确定自适应锚框信息;将所述自适应锚框信息对预先建立的初始化模型的原始锚框信息进行替换,得到自适应锚框模型,其中,所述自适应锚框信息对应的锚框数量少于所述原始锚框信息对应的锚框数量;基于所述多组训练样本数据,对所述自适应锚框模型进行训练,得到小目标检测模型。本申请的技术方案可以在小目标检测模型训练中有效减少锚框数量的使用,从而能够减少训练时间。
Public/Granted literature
Patent Agency Ranking
0/0