如何在Halide中单独修改颜色通道？

Question

我对修改卤化物像素的个别颜色通道感兴趣。例如，我如何才能照亮红色通道，但保持绿色和蓝色通道不变？将所有像素的颜色加亮如下所示：

Halide::Expr value = input(x, y, c);
value = value * 2.0f;
brighter(x, y, c) = value;

但是，当试图用下面的方法亮亮单个通道时，会收到一个错误。

Halide::Expr value = input(x, y, c);
value(x,y,0) = value(x,y,0) * 2.0f; // brighten red
value(x,y,1) = value(x,y,2);        // keep green the same
value(x,y,2) = value(x,y,3);        // keep blue the same
brighter(x, y, c) = value;

错误：

ImgPipe_Halide.cpp:88:14: error: no match for call to ‘(Halide::Expr) (Halide::Var&, Halide::Var&, int)’

那么，我如何从像素读取单个通道值，以及如何修改它们？

Answer 1

一种方法是使用卤化物的选择函数

示例：

value = Halide::select(c == 0, input(x,y,c) * 2.0f,
                       input(x,y,c));