【问题标题】:Torch: max value of each channel in an rgb imageTorch:RGB图像中每个通道的最大值
【发布时间】:2017-01-29 09:38:57
【问题描述】:

我正在尝试获取 rgb 图像中每个通道的最大值。 目前,我正在像这样分别迭代每个通道:

r_max = rgb[1]:max()
g_max = rgb[2]:max()
b_max = rgb[3]:max()

有什么方法可以将其理解为 Torch 中的单个语句?有人可以帮我解决可能的问题吗?

【问题讨论】:

  • 不太好,但您可以移至一行:r_max,g_max,b_max = rgb[1]:max(), rgb[2]:max(), rgb[3]:max ()

标签: image-processing lua torch


【解决方案1】:

Max 采用可选的“维度”参数。 在通道以外的维度上使用它会给你一个 3x1x1 的张量,如果需要,你可以将其展平,例如

 rgb_max = rgb:max(2):max(3):reshape(3)

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2020-02-13
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2014-03-15
    • 1970-01-01
    相关资源
    最近更新 更多