1. 程式人生 > >深度學習:乳腺論文 end2end patch->whole image

深度學習:乳腺論文 end2end patch->whole image

論文:end to end training for whole image breast cancer screening using an all convolutional design 

資料集:CBIS-DDSM (CC+MLO)*(right +left) one person ==》 (753鈣化)+(891腫塊)  資料增擴 

   patch set 

  •  s1: 1 ROI +1背景  in the same image 
  • s10:ROI overlapping raito of 0.9,  10 patch and 10 background  in the same image 

                    

任務:patch訓練 (5類:背景,鈣化良性,鈣化惡性,腫塊良性,腫塊惡性)

           二分類:正常(0-2),異常(4-6) (3不參與分類)

定位:CAM 

網路結構: Vggnet /Resnet+two classification

一、訓練流程:

中心思想:patch 訓練分類器,再用whole image 去(訓練+測試)

1、預訓練 ImageNet

2、patch train :用掩碼獲取patch,將五種patch送入網路訓練(resize224*224) train 5分類 (instance label) 

3、image train : 三種方法: (image label)

  • 將 patch classifier 訓練的 fc 層去掉,top layer +conv->2分類
  • 加入heatmap +fc -->2分類
  • 加入heatmap+random forest classifer  -->2分類