如何用CNN輸出圖像？

Question

我正在嘗試使用 CNN 進行深度估計（這是我的最終目標），但我發現的一個問題是：我只是使用 CNN 進行圖像分類，例如使用“CIFAR-10”、“MNIST”、“Cats vs Dogs”等。為了進行深度估計，我需要輸出一個新圖像（NYUv2 數據集具有標記圖像）。 所以，我將輸入一個像 256x256x3 的圖像，並需要輸出另一個圖像，例如 228x228x3。

我需要做什么？ 我可以只做一段時間的卷積，然后減少特征圖並增加維度嗎？ 謝謝

obs：我使用的是 Tensorflow 2.0

Answer 1

我建議你使用一種UNet 。 這種架構有下采樣層，然后是上采樣層，以恢復到原始的空間維度。