問題陳述通過使用 Kaggle 的 MRI 數據集的圖像分割來預測和定位腦腫瘤。這是該系列的第二部分。如果你還沒有閱讀第一部分，我建議你訪問使用深度學習進行腦腫瘤檢測和定位：第1部分以更好地理解代碼，因為這兩個部分是相互關聯的。文章地址：https：／／mp．weixin．qq．com／s／vBsTsVvHjA0gtQy3X1wdmw我們在 ResNet50 上訓練了一個分類模型，該模型使用回調對腦部 MRI 是否有腫瘤進行分類以提高我們的性能。在這一部分，我們將訓練一個模型來使用圖像分割來定位腫瘤。

現在，讓我們開始實施第二部分，即構建分割模型來定位腫瘤。圖像分割的目標是在像素級別理解圖像。它將每個像素與某個類相關聯。圖像分割模型產生的輸出稱為圖像的蒙版。首先，從我們在上一部分創建的數據幀中選擇蒙版值為 1 的記錄，因為只有腫瘤存在，我們才能對其進行定位。＃ Get the dataframe containing MRIs which have masks associated with them．
brain＿df＿mask ＝ brain＿df［brain＿df［＇mask＇］ ＝＝ 1］
brain＿df＿mask．shape
輸出：（1373， 4）將數據拆分為訓練和測試數據集。首先，我們將整個數據拆分為訓練和驗證數據，然后將一半的驗證數據拆分為測試數據。from sklearn．model＿selection import train＿test＿split
X＿train， X＿val ＝ train＿test＿split（brain＿df＿mask， test＿size＝0．15）
X＿test， X＿val ＝ train＿test＿split（X＿val， test＿size＝0．5）
我們將再次使用DataGenerator 生成虛擬數據，即training＿generator 和validation＿generator。為此，我們將首先創建要傳遞到生成器的圖像和蒙版路徑的列表。train＿ids ＝ list（X＿train．image＿path）
train＿mask ＝ list（X＿train．mask＿path）
val＿ids ＝ list（X＿val．image＿path）
val＿mask＝ list（X＿val．mask＿path）
＃ Utilities file contains the code for custom data generator
from utilities import DataGenerator
＃ create image generators
training＿generator ＝ DataGenerator（train＿ids，train＿mask）
validation＿generator ＝ DataGenerator（val＿ids，val＿mask）
定義一個如下所示的方法 Resblock ，以在我們的深度學習模型中使用。模型中使用 Resblocks 以獲得更好的結果。這些塊只是一堆層。resblocks 的主要功能是在頂部學習殘差函數，而信息沿底部傳遞不變。def resblock（X， f）：
 ＃ make a copy of input
 X＿copy ＝ X
 X ＝ Conv2D（f， kernel＿size ＝ （1，1） ，strides ＝ （1，1），kernel＿initializer ＝＇he＿normal＇）（X）
 X ＝ BatchNormalization（）（X）
 X ＝ Activation（＇relu＇）（X）
 X ＝ Conv2D（f， kernel＿size ＝ （3，3）， strides ＝（1，1）， padding ＝ ＇same＇， kernel＿initializer ＝＇he＿normal＇）（X）
 X ＝ BatchNormalization（）（X）
 X＿copy ＝ Conv2D（f， kernel＿size ＝ （1，1）， strides ＝（1，1）， kernel＿initializer ＝＇he＿normal＇）（X＿copy）
 X＿copy ＝ BatchNormalization（）（X＿copy）
 ＃ Adding the output from main path and short path together
 X ＝ Add（）（［X，X＿copy］）
 X ＝ Activation（＇relu＇）（X）
 return X
同樣，定義 upsample＿concat 方法來放大和連接傳遞的值。Upsampling 層是一個簡單的層，沒有權重，可以將輸入的維度加倍。def upsample＿concat（x， skip）：
 x ＝ UpSampling2D（（2，2））（x）
 merge ＝ Concatenate（）（［x， skip］）
 return merge
建立一個分割模型，添加下面顯示的層，包括上面定義的 resblock 和 upsample＿concat。input＿shape ＝ （256，256，3）
＃ Input tensor shape
X＿input ＝ Input（input＿shape）
＃ Stage 1
conv1＿in ＝ Conv2D（16，3，activation＝ ＇relu＇， padding ＝ ＇same＇， kernel＿initializer ＝＇he＿normal＇）（X＿input）
conv1＿in ＝ BatchNormalization（）（conv1＿in）
conv1＿in ＝ Conv2D（16，3，activation＝ ＇relu＇， padding ＝ ＇same＇， kernel＿initializer ＝＇he＿normal＇）（conv1＿in）
conv1＿in ＝ BatchNormalization（）（conv1＿in）
pool＿1 ＝ MaxPool2D（pool＿size ＝ （2，2））（conv1＿in）
＃ Stage 2
conv2＿in ＝ resblock（pool＿1， 32）
pool＿2 ＝ MaxPool2D（pool＿size ＝ （2，2））（conv2＿in）
＃ Stage 3
conv3＿in ＝ resblock（pool＿2， 64）
pool＿3 ＝ MaxPool2D（pool＿size ＝ （2，2））（conv3＿in）
＃ Stage 4
conv4＿in ＝ resblock（pool＿3， 128）
pool＿4 ＝ MaxPool2D（pool＿size ＝ （2，2））（conv4＿in）
＃ Stage 5 （Bottle Neck）
conv5＿in ＝ resblock（pool＿4， 256）
＃ Upscale stage 1
up＿1 ＝ upsample＿concat（conv5＿in， conv4＿in）
up＿1 ＝ resblock（up＿1， 128）
＃ Upscale stage 2
up＿2 ＝ upsample＿concat（up＿1， conv3＿in）
up＿2 ＝ resblock（up＿2， 64）
＃ Upscale stage 3
up＿3 ＝ upsample＿concat（up＿2， conv2＿in）
up＿3 ＝ resblock（up＿3， 32）
＃ Upscale stage 4
up＿4 ＝ upsample＿concat（up＿3， conv1＿in）
up＿4 ＝ resblock（up＿4， 16）
＃ Final Output
output ＝ Conv2D（1， （1，1）， padding ＝ ＂same＂， activation ＝ ＂sigmoid＂）（up＿4）
model＿seg ＝ Model（inputs ＝ X＿input， outputs ＝ output ）
編譯上面訓練的模型。這次我們將自定義優化器的參數。Focal tversky 是損失函數，tversky 是度量。＃ Utilities file also contains the code for custom loss function
from utilities import focal＿tversky， tversky
＃ Compile the model
adam ＝ tf．keras．optimizers．Adam（lr ＝ 0．05， epsilon ＝ 0．1）
model＿seg．compile（optimizer ＝ adam， loss ＝ focal＿tversky， metrics ＝ ［tversky］）
現在，你知道我們在分類器模型中使用的回調。我們將使用相同的方法來獲得更好的性能。最后，我們訓練我們的分割模型。＃ use early stopping to exit training if validation loss is not decreasing even after certain epochs．
earlystopping ＝ EarlyStopping（monitor＝＇val＿loss＇， mode＝＇min＇， verbose＝1， patience＝20）
＃ save the best model with lower validation loss
checkpointer ＝ ModelCheckpoint（filepath＝＂ResUNet－weights．hdf5＂， verbose＝1， save＿best＿only＝True）
model＿seg．fit（training＿generator， epochs ＝ 1， validation＿data ＝ validation＿generator， callbacks ＝ ［checkpointer， earlystopping］）
預測測試數據集的蒙版。這里，model是前面訓練的分類器模型，model＿seg是上面訓練的分割模型。from utilities import prediction
＃ making prediction
image＿id， mask， has＿mask ＝ prediction（test， model， model＿seg）
輸出將為我們提供圖像路徑、預測蒙版和類標簽。根據預測結果創建數據幀并與 image＿path 上的測試數據幀合并。＃ creating a dataframe for the result
df＿pred ＝ pd．DataFrame（｛＇image＿path＇： image＿id，＇predicted＿mask＇： mask，＇has＿mask＇： has＿mask｝）
＃ Merge the dataframe containing predicted results with the original test data．
df＿pred ＝ test．merge（df＿pred， on ＝ ＇image＿path＇）
df＿pred．head（）

正如你在輸出中看到的那樣，我們現在已將最終預測的蒙版合并到我們的數據幀中。最后，將原始圖像、原始蒙版和預測蒙版一起可視化，以分析我們的分割模型的準確性。count ＝ 0
fig， axs ＝ plt．subplots（10， 5， figsize＝（30， 50））
for i in range（len（df＿pred））：
 if df＿pred［＇has＿mask＇］［i］ ＝＝ 1 and count ＜ 5：
   ＃ read the images and convert them to RGB format
   img ＝ io．imread（df＿pred．image＿path［i］）
   img ＝ cv2．cvtColor（img， cv2．COLOR＿BGR2RGB）
   axs［count］［0］．title．set＿text（＂Brain MRI＂）
   axs［count］［0］．imshow（img）
   ＃ Obtain the mask for the image
   mask ＝ io．imread（df＿pred．mask＿path［i］）
   axs［count］［1］．title．set＿text（＂Original Mask＂）
   axs［count］［1］．imshow（mask）
   ＃ Obtain the predicted mask for the image
   predicted＿mask ＝ np．asarray（df＿pred．predicted＿mask［i］）［0］．squeeze（）．round（）
   axs［count］［2］．title．set＿text（＂AI Predicted Mask＂）
   axs［count］［2］．imshow（predicted＿mask）
   
   ＃ Apply the mask to the image ＇mask＝＝255＇
   img［mask ＝＝ 255］ ＝ （255， 0， 0）
   axs［count］［3］．title．set＿text（＂MRI with Original Mask （Ground Truth）＂）
   axs［count］［3］．imshow（img）
   img＿ ＝ io．imread（df＿pred．image＿path［i］）
   img＿ ＝ cv2．cvtColor（img＿， cv2．COLOR＿BGR2RGB）
   img＿［predicted＿mask ＝＝ 1］ ＝ （0， 255， 0）
   axs［count］［4］．title．set＿text（＂MRI with AI Predicted Mask＂）
   axs［count］［4］．imshow（img＿）
   count ＋＝ 1
fig．tight＿layout（）

輸出顯示我們的分割模型非常好地定位了腫瘤。做得好！此外，你可以嘗試向目前訓練的模型添加更多層并分析性能。還可以將類似的解決方案應用于其他問題陳述，因為圖像分割是當今非常感興趣的領域。