基于样式生成对抗网络的无人机巡检缺陷图像识别方法
摘要:
本发明提供了基于样式生成对抗网络的无人机巡检缺陷图像识方法,属于无人机图像识别技术领域,包括以下步骤:S1、建立无人机缺陷图像样本集;S2、构建基于样式的生成对抗网络模型,并利用步骤S1建立的无人机缺陷图像样本集对其进行训练;完成训练后利用该生成对抗网络模型生成模拟缺陷图像,并建立模拟缺陷图像样本集;S3、构建分类模型,并利用步骤S1建立的无人机缺陷图像样本集,和步骤S2建立的模拟缺陷图像样本集,对分类模型进行训练,直至其能精准识别缺陷类型;本发明能高效生成更多更高质量的模拟缺陷图像样本,从而训练提高分类模型,提高对无人机巡检图像中缺陷的识别率。
0/0