How to detect if a PNG image has transparent alpha channel or not using PIL?
img = Image.open('example.png', 'r') has_alpha = img.mode == 'RGBA'
With above code we know whether a PNG image has alpha channel not not but how to get the alpha value?
I didn’t find a ‘transparency’ key in img.info dictionary as described at PIL’s website
I’m using Ubuntu and zlib1g, zlibc packages are already installed.
To get the alpha layer of an RGBA image all you need to do is:
red, green, blue, alpha = img.split()
alpha = img.split()[-1]
And there is a method to set the alpha layer:
The transparency key is only used to define the transparency index in the palette mode (P). If you want to cover the palette mode transparency case as well and cover all cases you could do this
if img.mode in ('RGBA', 'LA') or (img.mode == 'P' and 'transparency' in img.info): alpha = img.convert('RGBA').split()[-1]
Note: The convert method is needed when the image.mode is LA, because of a bug in PIL.
You can get the alpha data out of whole image in one go by converting image to string with ‘A’ mode e.g this example get alpha data out of image and saves it as grey scale image 🙂
from PIL import Image imFile="white-arrow.png" im = Image.open(imFile, 'r') print im.mode == 'RGBA' rgbData = im.tostring("raw", "RGB") print len(rgbData) alphaData = im.tostring("raw", "A") print len(alphaData) alphaImage = Image.fromstring("L", im.size, alphaData) alphaImage.save(imFile+".alpha.png")
img.info is about the image as a whole — the alpha-value in an RGBA image is per-pixel, so of course it won’t be in
getpixel method of the image object, given a coordinate as argument, returns a tuple with the values of the (four, in this case) bands for that pixel — the tuple’s last value will then be A, the alpha value.
# python 2.6+ import operator, itertools def get_alpha_channel(image): "Return the alpha channel as a sequence of values" # first, which band is the alpha channel? try: alpha_index= image.getbands().index('A') except ValueError: return None # no alpha channel, presumably alpha_getter= operator.itemgetter(alpha_index) return itertools.imap(alpha_getter, image.getdata())
I tried this:
from PIL import Image import operator, itertools def get_alpha_channel(image): try: alpha_index = image.getbands().index('A') except ValueError: # no alpha channel, so convert to RGBA image = image.convert('RGBA') alpha_index = image.getbands().index('A') alpha_getter = operator.itemgetter(alpha_index) return itertools.imap(alpha_getter, image.getdata())
This returned the result that I was expecting. However, I did some calculation to determine the mean and standard deviation, and the results came out slightly different from imagemagick’s
Perhaps the conversion changed some of the values? I’m unsure, but it seems relatively trivial.