使用Python/PIL去除图像的背景颜色

Question

我一直在尝试让这个代码起作用，但遇到了很大的困难，因此我非常感激您的帮助。

使用下面的代码，我想将具有指定RGB值的特征更改为白色，并将图像中的所有其他特征更改为黑色（即从图像中提取特征）。不幸的是，虽然我可以很好地制作希望“提取”的特征，但当我尝试删除背景颜色时（我一直在尝试使用

mask2 = ((red != r1) & (green != g1) & (blue != b1))
data[:,:,:4][mask2] = [rb, gb, bb, ab]

但是似乎选择了除了红色==r1或绿色==g1等以外的所有像素，留下了一个相当“嘈杂”的背景图像。）有人知道一种可以从指定的RGB值中提取那些像素的方法，或者重新着色背景像素的更好方法吗？

谢谢

import numpy as np
from PIL import Image

im = Image.open('/home/me/nh09sw.tif')
im = im.convert('RGBA')
data = np.array(im)

r1, g1, b1 = 246, 213, 139 # Original value
rw, gw, bw, aw = 255, 255, 255, 255 # Value that we want to replace features with
rb, gb, bb, ab = 0, 0, 0, 255 #value we want to use as background colour

red, green, blue, alpha = data[:,:,0], data[:,:,1], data[:,:,2], data[:,:,3]

mask = ((red == r1) & (green == g1) & (blue == b1))
data[:,:,:4][mask] = [rw, gw, bw, aw]

im = Image.fromarray(data)

im.save('/home/me/nh09sw_recol.tif')

- hansolo

Answer 1

使用np.all()沿第三个轴进行比较。

import numpy as np
from PIL import Image

im = Image.open('my_file.tif')
im = im.convert('RGBA')
data = np.array(im)
# just use the rgb values for comparison
rgb = data[:,:,:3]
color = [246, 213, 139]   # Original value
black = [0,0,0, 255]
white = [255,255,255,255]
mask = np.all(rgb == color, axis = -1)
# change all pixels that match color to white
data[mask] = white

# change all pixels that don't match color to black
##data[np.logical_not(mask)] = black
new_im = Image.fromarray(data)
new_im.save('new_file.tif')

- wwii

---

灰度图像怎么样？ - alessiosavi

@alessiosavi - 相同的概念，但使用单个灰度值来代替r、g、b值作为掩码和替换。https://numpy.org/doc/stable/user/basics.indexing.html - wwii

@alessiosavi - 使用numpy将阈值以下的数字替换为零，还有其他人搜索“numpy灰度替换值”。 - wwii