python-3.x 转换rgb numpy图像到rgb和相应的索引值的列表

qyswt5oh  于 2023-02-01  发布在  Python
关注(0)|答案(1)|浏览(145)

我有一个表示rgb图像的numpy矩阵。它的形状是(n,m,3)n行,m列和3个通道。我想把它转换成rgb值列表沿着相应的索引。
我可以转换为RGB值列表,但我试图有行和列索引旁边以及。
我们可以只对rgb值做这样的操作。
flat_image = np.reshape(image, [-1,3]) # shape = [mxn, 3]
添加行号和列号后,形状应为[mxn,3+2]
因此,平面图像中的前三列表示RGB,第四列表示来自原始图像阵列的行号,第五列表示来自原始图像阵列的列号。

2lpgd968

2lpgd9681#

可以使用numpy.indices构造行/列索引,然后将其与flat_image连接

indices = np.indices(image.shape[:-1])

result = np.concatenate([flat_image, indices], axis=-1)

相关问题