基于YOLOX和分类回归损失目标识别方法
摘要:
本发明公开了一种基于YOLOX和分类回归损失目标识别方法,S1、获取待检测目标的相关高质量图像;S2、对高质量图像中的目标进行标注;S3、将数据集划分为训练数据集和验证数据集;S4、将损失函数中的回归损失函数进行分类处理,得到最终的损失函数;S5、将最终的损失函数应用于YOLOX目标检测算法中;S6、将训练数据集输入到目标检测算法中进行训练;S7、达到预设训练次数后,取训练的最后3次模型在测试数据集上进行分析,得到最优目标检测算法模型;S8、将最优目标检测算法模型用于识别待检测图像。本发明能够增加回归效果较差的检测框的损失值,让模型更加关注回归效果较差样本,从而提高目标检测任务的精度。
0/0