目标检测网络模型的训练方法、目标检测方法及相关设备
Abstract:
本申请公开了一种目标检测网络模型的训练方法、目标检测方法及相关设备,涉及计算机技术领域。方法包括:获取电网图像数据集;对电网图像数据集进行扩增处理,得到扩增电网图像数据集;确定扩增电网图像数据集中各电网图像的多标签信息以及与多标签信息相对应的第一损失,多标签信息包括电网图像中各无缺陷设备的标签;根据区域建议网络和用于预测目标类别和相对位置的R‑CNN头构建目标检测网络模型;根据区域建议网络和R‑CNN头的损失和第一损失确定总损失函数;将扩增电网图像数据集输入到目标检测网络模型中,使用总损失函数对目标检测网络模型进行训练,直到总损失函数收敛,得到目标检测网络模型。本申请可以提高目标检测的准确性和鲁棒性。
Patent Agency Ranking
0/0