[發明專利]圖像分割方法、裝置、電子設備及存儲介質

有效

• 申請號: 202111149022.0
• 申請日: 2021-09-28
• 公開（公告）號: CN113689435B
• 公開（公告）日: 2023-06-20
• 發明（設計）人: 司世景;王健宗
• 申請（專利權）人: 平安科技（深圳）有限公司
• 主分類號: G06T7/10
• 分類號: G06T7/10;G06V10/80;G06V10/82;G06V10/46;G06N3/0455;G16H30/20
• 代理公司: 廣州三環專利商標代理有限公司 44202
• 代理人: 熊永強

1.一種圖像分割方法，其特征在于，所述方法包括以下步驟：

對原始圖像進行特征提取，得到第一特征圖；

在對所述第一特征圖進行下采樣的過程中執行至少一次編碼操作，得到至少一個第二特征圖，所述至少一個第二特征圖與所述至少一次編碼操作一一對應；

將所述第一特征圖和所述至少一個第二特征圖進行稠密連接，得到融合特征圖；

基于所述融合特征圖對所述原始圖像中的目標對象進行分割，得到目標對象分割圖像；

所述至少一次編碼操作中的每次編碼操作，包括以下步驟：

采用殘差塊對下采樣得到的待編碼特征圖進行特征提取，得到第三特征圖；

將所述第三特征圖轉換為待編碼向量序列；

采用Transformer編碼器對所述待編碼向量序列進行編碼，得到編碼向量序列；

將所述編碼向量序列轉換為所述至少一個第二特征圖中的第二特征圖；

所述至少一個第二特征圖中的第二特征圖的尺度不同，所述第一特征圖與所述至少一個第二特征圖中的第二特征圖的尺度不同，所述將所述第一特征圖和所述至少一個第二特征圖進行稠密連接，得到融合特征圖，包括以下步驟：

按照至少一次解碼操作中每次解碼操作的輸入尺度，對所述第一特征圖和所述至少一個第二特征圖進行尺度歸一化處理；

將經過尺度歸一化處理的所述第一特征圖和所述至少一個第二特征圖合并，得到至少一個融合特征圖；

所述至少一個融合特征圖為所述至少一次解碼操作的輸入，所述至少一個融合特征圖與所述至少一次解碼操作一一對應，所述基于所述融合特征圖對所述原始圖像中的目標對象進行分割，得到目標對象分割圖像，包括以下步驟：

對所述至少一個融合特征圖中的第一個融合特征圖執行第一次解碼操作，得到目標解碼特征圖；

對所述目標解碼特征圖進行上采樣，得到第一個待合并特征圖；

重復對所述至少一個融合特征圖中的所述融合特征圖執行解碼操作，以及對解碼操作得到的解碼特征圖進行上采樣的步驟，直至所述至少一次解碼操作執行完畢，得到與所述第一特征圖尺度相同且未經過上采樣的第N個待合并特征圖；

基于所述第一個待合并特征圖至所述第N個待合并特征圖對所述原始圖像中的目標對象進行分割，得到所述目標對象分割圖像；

所述至少一次解碼操作中的每次解碼操作，包括以下步驟：

將所述融合特征圖轉換為待解碼向量序列；

采用Transformer解碼器對所述待解碼向量序列進行解碼，得到解碼向量序列；

將所述解碼向量序列轉換為第四特征圖；

采用殘差塊對所述第四特征圖進行特征提取，得到解碼特征圖。

2.根據權利要求1所述的方法，其特征在于，所述在對所述第一特征圖進行下采樣的過程中執行至少一次編碼操作，得到至少一個第二特征圖，包括以下步驟：

對所述第一特征圖進行下采樣，得到目標待編碼特征圖；

對所述目標待編碼特征圖執行第一次編碼操作，得到所述至少一個第二特征圖中的第一個第二特征圖；

重復對上一次編碼操作得到的第二特征圖進行下采樣，以及對下采樣得到的待編碼特征圖執行本次編碼操作的步驟，直至所述至少一次編碼操作執行完畢，得到所述至少一個第二特征圖中的第N個第二特征圖，N為大于等于1的整數；

由所述第一個第二特征圖至所述第N個第二特征圖組成所述至少一個第二特征圖。