首页/ 北京文安智能技术股份有限公司/ 专利详情

图像数据集处理方法和目标检测模型训练方法

基本信息

申请号 CN202110555366.5 申请日 -
公开（公告）号 CN113191451A 公开（公告）日 2021-07-30
申请公布号 CN113191451A 申请公布日 2021-07-30
分类号 G06K9/62（2006.01）I;G06N20/00（2019.01）I 分类 计算；推算；计数;
发明人 陈映;曹松;任必为 申请（专利权）人 北京文安智能技术股份有限公司
代理机构 - 代理人 -
地址 100094北京市海淀区丰豪东路9号院2号楼5单元801
法律状态 -

摘要

摘要 本发明提供了一种图像数据集处理方法和一种目标检测模型训练方法，其中，图像数据集处理方法包括统计图像数据集内样本图像中属于某类别的目标对象及其像素尺寸，在像素尺寸‑数量曲线上设定检测区域，绘制与平均数量值与像素尺寸‑数量曲线的交点；沿像素尺寸‑数量曲线分出多个区间，计算各区间对应的缩放概率集，根据缩放概率集按预设的算法步骤判定对各区间内的目标对象的像素尺寸缩放的处理方式。本发明解决了现有技术中的训练后的目标检测模型会在对不同数量的各类像素尺寸的目标对象检测时，存在明显的偏重误差以及不同的检测精度，进而造成训练后的目标检测模型的鲁棒性很差的问题。