2

我有一个带有 rgb 像素的图像数组,所以该数组有维度(image_count,height,width,3)

我正在通过我在侧面计算的随机量调整单个像素,形状为:(image_count,3)

现在,我有一个解释循环来为单个像素添加调整。


如何广播此添加,以便对每个像素执行以下操作:

image_set[i,h,w,:] += pixel_adj[i,:]
4

1 回答 1

2

None/np.newaxis引入具有适当尺寸的新轴pixel_adj以使其成为4D,就像image_set然后执行求和调整以引入broadcasting,像这样 -

image_set += pixel_adj[:,None,None,:]
于 2016-04-24T16:10:43.953 回答