[發明專利]管腔分割模型的訓練方法、裝置、電子設備及存儲介質

在審

• 申請號: 202211150263.1
• 申請日: 2022-09-21
• 公開（公告）號: CN115423826A
• 公開（公告）日: 2022-12-02
• 發明（設計）人: 黃星勝;馬駿;鄭凌霄;蘭宏志
• 申請（專利權）人: 深圳睿心智能醫療科技有限公司
• 主分類號: G06T7/11
• 分類號: G06T7/11;G06T7/187;G06T7/00;G06T5/50
• 代理公司: 北京品源專利代理有限公司 11332
• 代理人: 嚴慧

1.一種管腔分割模型的訓練方法，其特征在于，包括：

獲取管腔樣本圖像，以及與所述管腔樣本圖像對應的初始管腔標注圖像；

基于所述初始管腔標注圖像對應的管腔中心線，對初始管腔標注圖像進行狹窄檢測，確定管腔狹窄段；

對所述管腔狹窄段進行填充，得到狹窄填充圖像；

將所述狹窄填充圖像與所述初始管腔標注圖像進行合并，得到目標管腔標注圖像；

基于所述管腔樣本圖像和所述管腔樣本圖像對應的目標管腔標注圖像，對待訓練的管腔分割模型進行迭代訓練，得到訓練完成的管腔分割模型。

2.根據權利要求1所述的方法，其特征在于，在所述獲取管腔樣本圖像，以及與所述管腔樣本圖像對應的初始管腔標注圖像之后，所述方法還包括：

對于所述初始管腔標注圖像對應的管腔中心線中任一中心線點，基于所述中心線點至所述管腔中心線的起點的距離更新所述初始管腔標注圖像的管腔標簽值，得到管腔標簽優化圖像；

相應的，所述將所述狹窄填充圖像與所述初始管腔標注圖像進行合并，得到目標管腔標注圖像，包括：

將所述狹窄填充圖像與所述管腔標簽優化圖像進行合并，得到目標管腔標注圖像。

3.根據權利要求2所述的方法，其特征在于，所述基于所述中心線點至所述管腔中心線的起點的距離更新所述初始管腔標注圖像的管腔標簽值，得到管腔標簽優化圖像，包括：

將所述中心線點至所述管腔中心線的起點的距離，在預先設置的距離-標簽匹配關系中進行匹配，得到距離優化標簽值；

基于所述距離優化標簽值更新所述初始管腔標注圖像的管腔標簽值，得到管腔標簽優化圖像。

4.根據權利要求1所述的方法，其特征在于，所述基于所述初始管腔標注圖像對應的管腔中心線，對初始管腔標注圖像進行狹窄檢測，確定管腔狹窄段，包括：

對于所述初始管腔標注圖像對應的管腔中心線中任一中心線點，確定所述中心線點在所述初始管腔標注圖像中對應的管腔直徑；

對所述管腔中心線上所有中心線點對應的管腔直徑進行線性擬合處理，得到各中心線點對應的估計管腔直徑；

基于所述中心線點在所述初始管腔標注圖像中對應的管腔直徑與所述估計管腔直徑確定至少一個管腔狹窄段。

5.根據權利要求4所述的方法，其特征在于，所述基于所述中心線點在所述初始管腔標注圖像中對應的管腔直徑與所述估計管腔直徑確定至少一個管腔狹窄段，包括：

對于任一管腔狹窄段的確定步驟，包括：若多個連續中心線點對應的管腔直徑與估計管腔直徑的差值滿足預先設置的管腔狹窄條件，則將所述多個連續中心線點對應的管腔部分確定為管腔狹窄段。

6.根據權利要求1所述的方法，其特征在于，所述對所述管腔狹窄段進行填充，得到狹窄填充圖像，包括：

獲取所述管腔狹窄段的起始位置和終止位置；

基于所述管腔狹窄段的起始位置和終止位置，將所述管腔狹窄段的管腔直徑填充至估計管腔直徑，得到狹窄填充圖像。

7.根據權利要求6所述的方法，其特征在于，在所述得到狹窄填充圖像之后，還包括：

對于管腔狹窄段內任一中心線點，將所述管腔狹窄段內中心線點至所述管腔中心線的起點的距離，在預先設置的距離-標簽匹配關系中進行匹配，得到狹窄標簽值；

基于所述狹窄標簽值更新所述狹窄填充圖像的管腔標簽值，得到更新后的狹窄填充圖像。

8.根據權利要求1-7中任一項所述的方法，其特征在于，在所述得到訓練完成的管腔分割模型之后，還包括：

獲取至少一張待分割圖像；

將所述待分割圖像輸入至所述訓練完成的管腔分割模型中，得到與所述待分割圖像對應的預測分割圖像；

對所述預測分割圖像進行最大連通區域處理，得到最大連通區域圖像；

對所述最大連通區域圖像進行斑塊去除處理，得到目標分割圖像。