[發明專利]基于殘差unet的圖像分割算法

在審

• 申請號: 202310303167.4
• 申請日: 2023-03-27
• 公開（公告）號: CN116229078A
• 公開（公告）日: 2023-06-06
• 發明（設計）人: 王超;陳宏銘
• 申請（專利權）人: 蘇州芯運旺電子科技有限公司
• 主分類號: G06V10/26
• 分類號: G06V10/26;G06V10/764;G06V10/774;G06V10/82;G06N3/0455;G06N3/08
• 代理公司: 蘇州高專知識產權代理事務所(特殊普通合伙) 32474
• 代理人: 冷泠

1.一種基于殘差unet的圖像分割算法，其特征在于：步驟包括：

1)圖像分割：

通過Y_in∈R^2×H×W表示三維的圖像數據，進行分割產生N維的分割數據Y_class∈R^N×H×W，再通過分類器進行上色顯示得到Y_out∈R^N×H×W；

H和W代表了圖像數據的高度和寬度，N表示圖片中的種類；圖像分割模型的建模為：

Y_class＝F_bsekbane(Y_in)???????????(1)

Y_out＝F_classifier(Y_class)????????(2)

2)殘差運算：

設置跨尺度殘差傳遞模塊，并進行對應跨尺度殘差傳遞模塊的訓練，采用交叉熵損失函數用于約束模型的最優目標；

公式化表述如下：

y_i表示真實概率標簽，表示預測結果分布。

2.根據權利要求1所述的一種基于殘差unet的圖像分割算法，其特征在于：圖像分割模型采用U-Net架構作為骨干，U-Net架構可以捕獲多尺度特征，包含更多的底層信息。

3.根據權利要求2所述的一種基于殘差unet的圖像分割算法，其特征在于：跨尺度殘差傳遞模塊具有良好的信息交互功能，同時提供兩個尺度的信息，進而提取到主干網絡外更多的細節信息。

4.根據權利要求3所述的一種基于殘差unet的圖像分割算法，其特征在于：跨尺度殘差傳遞模塊使用上采樣和下采樣的差值傳遞差異信息；在殘差運算中，特征圖會傳入下采樣模塊中，以獲得較低分辨率大小的圖像。

5.根據權利要求4所述的一種基于殘差unet的圖像分割算法，其特征在于：使用卷積來對低分辨率的圖像進行特征提取得到低分辨率的特征圖。

6.根據權利要求5所述的一種基于殘差unet的圖像分割算法，其特征在于：將低分辨率的特征圖通過上采樣模塊將得到高分辨率的特征圖，最后將兩個高分辨率的圖殘差運算的結果輸入到解碼器中。