如何用CNN输出图像？

Question

我正在尝试使用 CNN 进行深度估计（这是我的最终目标），但我发现的一个问题是：我只是使用 CNN 进行图像分类，例如使用“CIFAR-10”、“MNIST”、“Cats vs Dogs”等。为了进行深度估计，我需要输出一个新图像（NYUv2 数据集具有标记图像）。 所以，我将输入一个像 256x256x3 的图像，并需要输出另一个图像，例如 228x228x3。

我需要做什么？ 我可以只做一段时间的卷积，然后减少特征图并增加维度吗？ 谢谢

obs：我使用的是 Tensorflow 2.0

Answer 1

我建议你使用一种UNet 。 这种架构有下采样层，然后是上采样层，以恢复到原始的空间维度。