变电站电力设备图像分割模型、方法及训练方法
摘要:
本申请实施例提供变电站电力设备图像分割模型、方法及训练方法,包括:编码器、第一特征传输模块、第二特征传输模块和解码器;编码器,用于提取变电站电力设备扫描图像的多层次特征,形成不同层次的多个特征图;第一特征传输模块,用于对多个特征图中的长距离相关特征进行特征提取,捕捉不同区域特征之间的关联性,得到第一特征图;第二特征传输模块,用于对多个特征图分别进行特征融合和多尺度卷积操作,得到多个融合特征图;解码器,用于对第一特征图和融合特征图进行叠加及上采样操作,提取不同粒度的图像局部特征,得到扫描图像的分割结果;能够提高变电站复杂场景下电力设备图像的分割精度;适用于电力设备图像识别的技术领域。
0/0