对透明背景的白色背景使用PIL Python

时间:2011-03-19 23:18:31

标签: python python-imaging-library

如何使用PIL在透明背景中转换png或jpg图像的所有白色背景和白色元素?

1 个答案:

答案 0 :(得分:5)

使用numpy,以下内容使白色区域透明。您可以更改thresholddist来控制“white-ish”的定义。

import Image
import numpy as np

threshold=100
dist=5
img=Image.open(FNAME).convert('RGBA')
# np.asarray(img) is read only. Wrap it in np.array to make it modifiable.
arr=np.array(np.asarray(img))
r,g,b,a=np.rollaxis(arr,axis=-1)    
mask=((r>threshold)
      & (g>threshold)
      & (b>threshold)
      & (np.abs(r-g)<dist)
      & (np.abs(r-b)<dist)
      & (np.abs(g-b)<dist)
      )
arr[mask,3]=0
img=Image.fromarray(arr,mode='RGBA')
img.save('/tmp/out.png')

代码很容易修改,只有RGB值(255,255,255)变为透明 - 如果这是你真正想要的。只需将mask更改为:

即可
mask=((r==255)&(g==255)&(b==255)).T