[发明专利]基于类别平衡采样的无人机巡检航拍图片批处理训练方法

摘要

本发明公开了一种基于类别平衡采样的无人机巡检航拍图片数据样本批处理训练方法，步骤如下：S1、按照正常、缺陷类别对无人机巡检航拍图进行排序，分别计算出缺陷、正常类别数据样本的数量；S2、根据正常类别样本数分别对正常类与缺陷类生成随机排列的列表，将随机列表的随机数对各自对应的类别样本数取余，所得的结果作为对应的索引值；S3、根据索引值提取对应类别的图片数据以生成图片随机列表；S4、将正常类别与缺陷类别的图像随机列表结合在一起，并且打乱顺序，使得缺陷类别数据样本数与正常类别样本数相同；S5、根据上述图像最终列表批处理训练模型，当本轮训练结束后，重复上述操作进行下一轮训练。

主权项

1.一种基于类别平衡采样的无人机巡检航拍图片数据样本批处理训练方法，其特征在于，所述的训练方法包括以下步骤：S1、按照类别对图片排序，统计各个类别包含数据样本的数量；S2、根据某类最多样本数对各个类别生成随机列表，使用随机列表中的数值对相应类别样本数取余，得到相应的索引值；S3、根据所述的索引值从属于该类别的数据样本中提取图片，将获取的图片生成各自类别的图片随机列表；S4、将以上每一类别的图片随机列表整合在一起，然后打乱顺序，得到最终的图片列表；S5、使用最终的图片列表进行批处理训练，直至本轮训练结束，重复上述操作进行下一轮训练。