python PIL 和opencv检测图片是否包含透明通道

最新推荐文章于 2024-04-26 17:19:48 发布

原创最新推荐文章于 2024-04-26 17:19:48 发布 · 7.2k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#opencv #python #计算机视觉 #PIL

图像/音频/视频处理专栏收录该内容

25 篇文章

订阅专栏

这篇博客对比了PIL和OpenCV库在检测图像透明度方面的准确性。通过示例代码，作者展示了如何使用PIL的getcolors()方法和OpenCV的imread()函数来检查PNG图像的透明度通道。结果显示，PIL在这项任务上的表现更准确，而OpenCV在某些情况下可能不准确。

PIL检测更加准确：

from PIL import Image


def has_transparency(img):
    if img.mode == "P":
        transparent = img.info.get("transparency", -1)
        for _, index in img.getcolors():
            if index == transparent:
                return True
    elif img.mode == "RGBA":
        extrema = img.getextrema()
        if extrema[3][0] < 255:
            return True

    return False


if __name__ == "__main__":
    # extract_alpha("./13.png")
    image = Image.open("./Female_Model_Nurse.png")
    print(has_transparency(image))

opencv检测不准确：

import cv2

def alpha_to_image(alpha, image_path):
    """
    单独将 alpha 保存成图像
    :param alpha:
    :param image_path:
    :return:
    """
    cv2.imwrite(image_path, alpha)


def extract_alpha(image_path):
    """
    提取图片的 alpha 通道
    :param image_path:
    :return:
    """
    # im = cv2.imread(image_path, cv2.IMREAD_UNCHANGED)  # 顾名思义，读入完整图片，包括alpha通道
    # im = cv2.imread(image_path, cv2.IMREAD_COLOR)  # cv2.IMREAD_COLOR：默认参数，读入一副彩色图片，忽略alpha通道
    # im = cv2.imread(image_path, cv2.IMREAD_GRAYSCALE)  # cv2.IMREAD_GRAYSCALE：读入灰度图片
    im = cv2.imread(image_path)
    # 颜色空间转换?
    # if im.shape[-1] == 3:
    #     img = cv2.cvtColor(im, cv2.COLOR_BGR2BGRA)

    im_shape = im.shape
    print(f"im_shape = {im_shape}")

    # 显示图片查看
    # 显示图片查看
    cv2.imshow('imshow', im)
    cv2.waitKey(1000)
    cv2.destroyAllWindows()
    # 显示图片查看

    # # 显示图片查看

    #
    # if len(im_shape) != 4:
    #     print("此图片不包含 alpha 通道")
    # alpha = im[3]
    # # 此时获取的就是 alpha 通道
    # return alpha


if __name__ == "__main__":
    pass
    # print(extract_alpha('./11.png'))
    # print(extract_alpha('./bb.png'))
    print(extract_alpha('./Female_Model_Nurse.png'))