分离 Alpha 通道

从 PNG 图片中分离出 alpha 通道

# -*- coding: utf-8 -*-
# @Time : 2022/6/23 15:38
# @Author : lc
# @File : get_mask.py
# @Software: PyCharm
# @Function: Separate out the alpha channel

import os
import cv2


def get_pha(png_dir):
    file_list = os.listdir(png_dir)
    for file in file_list:
        file_path = os.path.join(png_dir, file)
        img_list = os.listdir(file_path)
        for img in img_list:
            img_path = os.path.join(file_path, img)
            save_path = img_path.replace('fgr_png', 'mask')
            img = cv2.imread(img_path, flags=-1)
            print(img_path.split('\\')[-1])
            b, g, r, alpha = cv2.split(img)
            cv2.imwrite(save_path, alpha)
            # cv2.imshow('alpha', alpha)
            # cv2.waitKey(0)
        print('{} saved !'.format(file))


if __name__ == "__main__":
    png_dir = r"./fgr_png"
    get_pha(png_dir)