我有一个带有 rgb 像素的图像数组,所以该数组有维度(image_count,height,width,3)
我正在通过我在侧面计算的随机量调整单个像素,形状为:(image_count,3)
。
现在,我有一个解释循环来为单个像素添加调整。
如何广播此添加,以便对每个像素执行以下操作:
image_set[i,h,w,:] += pixel_adj[i,:]
我有一个带有 rgb 像素的图像数组,所以该数组有维度(image_count,height,width,3)
我正在通过我在侧面计算的随机量调整单个像素,形状为:(image_count,3)
。
现在,我有一个解释循环来为单个像素添加调整。
如何广播此添加,以便对每个像素执行以下操作:
image_set[i,h,w,:] += pixel_adj[i,:]
None/np.newaxis
引入具有适当尺寸的新轴pixel_adj
以使其成为4D
,就像image_set
然后执行求和调整以引入broadcasting
,像这样 -
image_set += pixel_adj[:,None,None,:]