一种变电设备缺陷图像数据扩充与数据清洗方法
摘要:
本发明提供了一种变电设备缺陷图像数据扩充与数据清洗方法,包括:利用预设数据集对风格迁移网络进行训练,得到训练好的网络权重参数;遍历原始变电设备缺陷图像和弱光场景图像,并利用加载所述网络权重参数的所述风格迁移网络,根据所述原始变电设备缺陷图像和弱光场景图像对缺陷图像数据进行扩充,得到低光照变电设备缺陷图像数据,并验证风格迁移效果;滤除所述低光照变电设备缺陷图像数据中的低质量生成图像,并依托对抗验证方法利用分类模型筛选出与真实图像集合样本分布近似的生成图像,得到最终扩充的低光照变电设备缺陷图像集合。本发明有助于解决深度学习模型在弱光场景泛化性弱的问题。
0/0