科研之友
微信
新浪微博
Facebook
分享链接
摘要
本发明涉及图像处理技术领域,公开了一种基于软标签的目标检测样本平衡方法,所述方法包括：获取输入图像的目标的个数以及候选框的个数；计算每一个目标与每一个候选框的第一IOU值；对所述第一IOU值按列取最大值,得到第二IOU值,并获取所述第二IOU值在所述第一IOU值中的行索引值；对每一个目标,从所述行索引值中获取负责该目标的所有目标候选框,基于软标签获取每一个目标候选框的标签值；根据每一个目标候选框的标签值以及预先构建的损失函数获取每一个目标候选框的预测概率；根据所述预测概率输出目标检测框。本发明提供的一种基于软标签的目标检测样本平衡方法,能够对目标检测过程中的样本数量进行平衡,提高检测结果的准确率。
- 单位
