我有一个RGB彩色图像遮罩mask_color
,形状(4,4,3)。如何快速地将所有黑色像素[0,0,0]
转换为白色[255,255,255]
,而不使用任何循环,不使用额外的包,最好是NumPy方式
mask_color = np.array([
[
[0,0,0],
[128,0,255],
[0,0,0],
[0,0,0]
],
[
[0,0,0],
[0,0,0],
[0,0,0],
[0,0,0]
],
[
[0,0,0],
[50,128,0],
[0,0,0],
[0,0,0]
],
[
[0,0,0],
[0,0,0],
[245,108,60],
[0,0,0]
]
])
plt.imshow(mask_color)
plt.show()
white_bg_mask_color = # do something
plt.imshow(white_bg_mask_color)
plt.show()
您也可以像下面这样使用布尔索引
您可以使用np.where:
相关问题 更多 >
编程相关推荐