一种加速目标检测算法训练的数据集处理方法

基本信息

申请号 CN202110853633.7 申请日 -
公开(公告)号 CN113313213A 公开(公告)日 2021-08-27
申请公布号 CN113313213A 申请公布日 2021-08-27
分类号 G06K9/62 分类 计算;推算;计数;
发明人 王纵驰;王建兴;刘健;臧媛媛;王培;于佳耕 申请(专利权)人 航天神舟智慧系统技术有限公司
代理机构 北京中知法苑知识产权代理有限公司 代理人 李明;赵吉阳
地址 100088 北京市海淀区马甸路2号中航油大厦
法律状态 -

摘要

摘要 一种加速目标检测算法训练的数据集处理方法,包括:(1)获取锚框类目标检测算法所需的训练用数据集;(2)数据集数据分析,提取数据集中所有标注框的特征作为待处理数据的划分依据;(3)数据集聚类,使用聚类算法,按照小、中、大三个尺度对特征进行聚类划分;(4)数据集划分;(5)模型训练,构成三阶训练体系,并按照先验知识对批处理数据簇进行分步训练;提高了深度学习数据集的训练效率,使用灵活方便,极大的加速了训练进度从而缩短训练时间。