如何使用CNN输出图像？

Question

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

我需要做什么？我可以暂时进行卷积，然后减少特征图并增加维数吗？谢谢

obs：我正在使用Tensorflow 2.0

Answer 1

我建议您使用UNet类型。这种架构具有下采样层，其后是上采样层，以返回到原始空间尺寸。