科研之友
微信
新浪微博
Facebook
分享链接
摘要
本发明公开了一种面向热成像行人检测的样本处理方法。所述方法包括以下步骤：对初始数据集划分训练集和测试集,训练初始主干网络,从初始训练集中依据尺寸和遮挡情况筛选获得标注行人样本集合,使用初始主干网络提取集合中样本特征；对行人特征聚类,生成行人样本的子类别标签；基于行人子类别标签统计行人类内分布,对行人子类别中少数类进行迭代增强,平衡行人样本的类内分布,生成平衡训练集；统计设计平衡损失函数,利用平衡训练集训练神经网络模型。本发明能够在不增加系统计算量的情况下优化行人检测数据集数据分布,提升所训练神经网络模型检测精度。
- 单位
