[發明專利]三維重建網絡的訓練方法及裝置、三維重建方法及裝置

有效

• 申請號: 202011321648.0
• 申請日: 2020-11-23
• 公開（公告）號: CN112435341B
• 公開（公告）日: 2022-08-19
• 發明（設計）人: 尹思源;李玉才;余航;張歡;王瑜;王少康;陳寬
• 申請（專利權）人: 推想醫療科技股份有限公司
• 主分類號: G06T17/20
• 分類號: G06T17/20;G06T7/11;G06N3/04;G06N3/08
• 代理公司: 北京布瑞知識產權代理有限公司 11505
• 代理人: 孟潭

1.一種三維重建網絡的訓練方法，其特征在于，包括：

獲取三維醫學圖像樣本對應的一張二維醫學圖像樣本，其中，所述二維醫學圖像樣本基于所述三維醫學圖像樣本而生成；

將所述二維醫學圖像樣本輸入三維重建網絡，獲得所述二維醫學圖像樣本對應的三維重建分割圖像，其中，所述三維重建分割圖像為重建的分割后的三維醫學圖像；

基于所述三維重建分割圖像和所述三維醫學圖像樣本上標記的三維分割數據，對所述三維重建網絡進行訓練，

其中，所述三維重建網絡包括編碼網絡、轉換網絡和解碼網絡，所述將所述二維醫學圖像樣本輸入三維重建網絡，獲得所述二維醫學圖像樣本對應的三維重建分割圖像，包括：

利用所述編碼網絡提取所述二維醫學圖像樣本的二維特征；

利用所述轉換網絡將所述二維醫學圖像樣本的二維特征轉換為三維特征；

利用所述解碼網絡將所述三維特征轉換為所述二維醫學圖像樣本對應的三維重建分割圖像；

所述基于所述三維重建分割圖像和所述三維醫學圖像樣本上標記的三維分割數據，對所述三維重建網絡進行訓練，包括：

將所述三維醫學圖像樣本上標記的三維分割數據作為三維分割標簽，根據所述三維重建分割圖像和所述三維分割標簽的差異，訓練所述三維重建網絡。

2.根據權利要求1所述的方法，其特征在于，所述基于所述三維重建分割圖像和所述三維醫學圖像樣本上標記的三維分割數據，對所述三維重建網絡進行訓練，包括：

基于三維重建分割圖像和所述三維醫學圖像樣本上標記的三維分割數據，對所述三維重建網絡和鑒別網絡進行交替對抗訓練。

3.根據權利要求2所述的方法，其特征在于，所述基于三維重建分割圖像和所述三維醫學圖像樣本上標記的三維分割數據，對所述三維重建網絡和鑒別網絡進行交替對抗訓練，包括：

將所述三維重建網絡生成的所述三維重建分割圖像輸入所述鑒別網絡，獲得第一損失值，其中，所述三維重建分割圖像的真偽標簽設置為假；

將所述三維醫學圖像樣本上標記的三維分割數據輸入所述鑒別網絡，獲得第二損失值，其中，所述三維醫學圖像樣本上標記的三維分割數據的真偽標簽設置為真；

根據所述第一損失值和所述第二損失值，訓練所述鑒別網絡；

將所述三維重建網絡生成的所述三維重建分割圖像輸入所述鑒別網絡，獲得第三損失值，其中，所述三維重建分割圖像的真偽標簽設置為真；

根據所述第三損失值，訓練所述三維重建網絡。

4.根據權利要求1至3中任一項所述的方法，其特征在于，所述三維醫學圖像樣本為三維CT肋骨圖像，所述二維醫學圖像樣本為二維DR肋骨圖像，所述三維重建分割圖像為重建的分割后的三維CT肋骨圖像。

5.一種三維重建方法，其特征在于，包括：

獲取待重建的二維醫學圖像；

將所述待重建的二維醫學圖像輸入根據權利要求1至4中任一項所述的三維重建網絡的訓練方法訓練得到的三維重建網絡，獲得所述待重建的二維醫學圖像對應的三維重建分割圖像，其中，所述三維重建分割圖像為重建的分割后的三維醫學圖像，

其中，所述三維重建網絡包括編碼網絡、轉換網絡和解碼網絡，所述將所述待重建的二維醫學圖像輸入根據權利要求1至4中任一項所述的三維重建網絡的訓練方法訓練得到的三維重建網絡，獲得所述待重建的二維醫學圖像對應的三維重建分割圖像，包括：

利用所述編碼網絡提取所述待重建的二維醫學圖像的二維特征；

利用所述轉換網絡將所述待重建的二維醫學圖像的二維特征轉換為三維特征；

利用所述解碼網絡將所述三維特征轉換為所述待重建的二維醫學圖像對應的三維重建分割圖像。