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