在Tensorflow中迭代张量的行和列

时间:2017-12-15 15:31:26

标签: image tensorflow threshold adaptive-threshold

我的项目的一部分是在图像上使用阈值内核。 阈值内核可能如下所示:

[50  100]
[150 200]

我想查看每组3x3像素(没有重叠),并使用我的内核对它们进行阈值处理。

例如,如果我有这个灰度图像:

[120 120 120 120]
[120 120 120 120]
[170 170 170 170]
[170 170 170 170]

然后在阈值处理后我得到这个图像:

[1 1 1 1]
[0 0 0 0]
[1 1 1 1]
[1 0 1 0]

我正在使用TensorFlow,我的网络需要不同形状的批次。

输入是:

data['input_tensor'] = tf.placeholder(tf.float32, shape=[None, None, None, 1], name='Input')

阈值内核的类型为:tf.Variable(),大小为5x5, 及其价值应该被学习!

我无法找到实现目标的方法。 我尝试迭代输入批处理,但其大小未知(仅在会话期间知道)。 我不想复制阈值内核,因为那时网络将尝试学习它的所有值(现在它太多了。)

有没有办法没有循环呢? 如果没有,我怎么能用循环呢?

感谢。

0 个答案:

没有答案