基于类别随机采样的架空输电线路缺陷检测数据样本批处理训练方法

作者：王凯; 王健; **; 周文青; 毕跃凡

来源：2018-11-23, 中国, CN201811407820.7.

本发明公开了一种基于类别随机采样的架空输电线路缺陷检测数据样本批处理训练方法,步骤如下：S1、无人机采集数据,确定每个数据样本对应的类别,按类别存放数据,随后统计每一类所包含样本的数量；S2、根据每一类样本包含的数量确定深度学习算法模型批处理数量Nbatch的数值；S3、为保证批处理数据中每个类别样本数量相同,从每一类中随机抽取Nbatch/n个数据样本；S4、将抽取到的数据组合成一组批处理数据,打乱顺序后用于模型批处理训练；S5、待此一批数据训练结束后,重复上述步骤S3和S4进行下一次批处理训练。本发明在数据层面保证了用于训练的每一批数据中各个类别样本数量相同,有效提高训练模型的泛化能力。

• 单位
  华南理工大学