tensorflow - 使用 tensorflow 创建图像的颜色直方图

Question

有没有一种巧妙的方法来计算图像的颜色直方图？也许是通过滥用 tf.histogram_summary 的内部代码？据我所知，这段代码不是很模块化，而是直接调用一些 C++ 代码。

提前致谢。

Answer 1

我会使用tf.unsorted_segment_sum ，其中“段 ID”是根据颜色值计算得出的，求和结果是一个 tf.ones 向量。请注意，tf.unsorted_segment_sum 可能更好地被认为是“桶总和”。它实现了 dest[segment] += thing_to_sum ——正是直方图所需的操作。

稍微伪代码(意味着我还没有运行这个):

binned_values = tf.reshape(tf.floor(img_r * (NUM_BINS-1)), [-1])
binned_values = tf.cast(binned_values, tf.int32)
ones = tf.ones_like(binned_values, dtype=tf.int32)
counts = tf.unsorted_segment_sum(ones, binned_values, NUM_BINS)

如果您想巧妙地构造“1”，使其看起来像红色的“100100...”、红色的“010010”，则可以一次性完成此操作，而不是通过拆分来分离 r、g 和 b 值。绿色等，但我怀疑它总体上会更慢，并且更难阅读。我只是按照您上面建议的方式进行拆分。