在图像上滑动2D模型以生成标签图像(非卷积)

我有一个从RGB图像样本中训练的模型,该模型以31×31像素区域作为输入,并为中心像素生成一个分类结果。

我想将这个模型应用于整张图像,以便有效地恢复每个像素的分类新图像。由于这不是卷积,我不确定在TensorFlow中这样做的首选方法是什么。

我知道可以通过将图像分解成许多小的张量来进行推断,但这样做似乎是巨大的浪费,因为每个像素会被复制961次。有没有办法避免这种情况?


回答:

将您的模型改造成一个全卷积神经网络,这样对于31x31的图像,它将生成一个单一标签,而对于62x62的图像,它将生成2x2的标签,依此类推。这将消除您在窗口方法中提到的冗余计算问题。

如果网络中有一个全连接层,可以用1x1核的卷积层来替换它。

Related Posts

L1-L2正则化的不同系数

我想对网络的权重同时应用L1和L2正则化。然而,我找不…

使用scikit-learn的无监督方法将列表分类成不同组别,有没有办法?

我有一系列实例,每个实例都有一份列表,代表它所遵循的不…

f1_score metric in lightgbm

我想使用自定义指标f1_score来训练一个lgb模型…

通过相关系数矩阵进行特征选择

我在测试不同的算法时,如逻辑回归、高斯朴素贝叶斯、随机…

可以将机器学习库用于流式输入和输出吗?

已关闭。此问题需要更加聚焦。目前不接受回答。 想要改进…

在TensorFlow中,queue.dequeue_up_to()方法的用途是什么?

我对这个方法感到非常困惑,特别是当我发现这个令人费解的…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注