1. 程式人生 > >keras U-Net實現

keras U-Net實現

程式碼位置

https://github.com/lsh1994/keras-segmentation

模型結構

我這裡用到了vgg16微調作為編碼器,讀者可以參照著自定義層對稱的編解碼結構。

U-Net網路處理輸入時進行了鏡面放大2倍,所以最終的輸入輸出縮小了2倍。此處直接在靠後的輸出上上取樣置原始影象大小。

在這裡插入圖片描述

訓練結果

在這裡插入圖片描述

在這裡插入圖片描述

在這裡插入圖片描述