一种基于改进的U2-Net的变电站环境感知的方法

    公开(公告)号:CN116168293A

    公开(公告)日:2023-05-26

    申请号:CN202310129535.8

    申请日:2023-02-17

    Abstract: 本发明公开一种基于改进的U²‑Net的变电站环境感知的方法,包括以下步骤:a、采集变电站数据集:获得变电站图像;b、标注数据集:按照景物类别对变电站图像标注标签,获得标注后的变电站图像;c、图像识别:采用改进的U²‑Net语义分割算法模型对标注后的变电站图像进行识别,输出与标签类别种类数量相同的分割图像。改进的U²‑Net语义分割算法模型的输出端的网络包括:encoder‑decoder模块,用于融合不同尺寸感受野的特征,捕获不同尺度的上下文信息;Focalloss损失函数,用于解决样本的非均衡性问题;GELU激活函数;网络输出层,使输出特征图上的每一个像素点都包含对应类的概率。本发明能够实时地、准确地识别出变电站物体信息的种类,对图像细节的分割效果好。

Patent Agency Ranking