【百戰GAN】GAN也可以拿來做圖像分割，看起來效果還不錯？

大家好，歡迎來到專欄《百戰GAN》，在這個專欄里，我們會進行算法的核心思想講解，代碼的詳解，模型的訓練和測試等內容。

作者&編輯 | 言有三

本文資源與生成結果展示

本文篇幅：5600字

背景要求：會使用Python和Pytorch

附帶資料：項目推薦，版本包括Pytorch+Tensorflow

同步平台：有三AI知識星球(一周內)

1 項目背景

生成對抗網絡如今在計算機視覺的很多領域中都被廣泛應用，圖像分割是一個非常基礎的任務，兩者的結合會擦出什麼樣的火花呢？這一期我們使用GAN完成圖像分割任務，本次需要做的準備工作包括：

(1) Linux系統或者windows系統，使用Linux效率更高。

(2) 安裝好的Pytorch，需要GPU進行訓練。

2 原理簡介

圖像分割任務輸入是一張圖像，輸出是一張與圖像大小相等的掩膜。要使用GAN來完成這個任務，也要滿足這樣的條件，這與之前介紹的輸入噪聲向量，輸出生成圖像有所不同，下面是整個框架示意圖：

該框架輸入圖x，經過生成器G之後得到分割結果G(x)，然後一邊將G(x)和x一起作為判別器的輸入，另一邊將真實標註y和x一起作為判別器的輸入，前者判別器輸出為fake，後者為real。