python PIL.Image 如何将图像自带的Alpha透明度去除，例如将'RGBA'

前言
在练习文生图大模型之前，网络练习数据是很重要的一步，
而在网络上获取的许多图片规格不一，给数据的同一处置惩罚和导入带来了不少麻烦。
其中有一种特殊环境，就是图像自带透明度（Alpha），通常在文件信息中就会有显示，好比下面这张图：

    

在这种环境下，如果直接用cv2或PIL将它转化为"RGB"模式，会导致得到的图片是纯玄色的。
 
解决办法
如果是由于透明度导致，那么解决的直接思路就是给整张图片加一块同等大小的白色背景，
这里用到的函数为：

1. PIL.Image.alpha_composite()

复制代码

 
在批量处置惩罚图片时，还需要加上判断该图片是否包含透明度的逻辑，
代码如下：

1. from PIL import Image
2. import os
3. from os.path import join as ospj
5. src_dir = ''
6. dst_dir = ''
8. for f in os.listdir(src_dir):
9.     src_path = ospj(src_dir, f)
10.     dst_path = ospj(dst_dir, f)
11.     if f.endswith('DS_Store'):
12.         continue
13.     else:
14.         img = img = Image.open(src_path)
15.         # 查看图片是否包含透明度
16.         if img.mode in ('RGBA', 'LA') or (img.mode == 'P' and 'transparency' in img.info):
17.             # 准备一张同等尺寸的白色背景
18.             background = Image.new('RGBA', img.size, (255, 255, 255))
19.             # 复合原图与背景图
20.             alpha_composite = Image.alpha_composite(background, img)
21.             # 将复合后的图片转为"RGB"模式并保存
22.             alpha_composite = alpha_composite.convert('RGB')
23.             alpha_composite.save(dst_path)
24.         else:
25.             img.save(dst_path)

复制代码

 
 
The End