从 PNG 图片中分离出 alpha 通道
# -*- coding: utf-8 -*-# @Time : /6/23 15:38# @Author : lc# @File : get_mask.py# @Software: PyCharm# @Function: Separate out the alpha channelimport osimport cv2def 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)
