from PIL import Image, ImageDraw, ImageFont import os text = "watermark" # 水印文字 font = ImageFont.truetype("arial.ttf", 36) # 字体和大小 path = r"C:\Users\Administrator\Desktop\0424特价直播样品\新建文件夹\新建文件夹" for filename in os.listdir(path): print(path,filename) if filename.endswith(".jpg") or filename.endswith(".png") or filename.endswith(".JPG") or filename.endswith(".PNG"): # 只处理jpg和png图片 # 处理每一张图片 # 打开图片 image = Image.open(os.path.join(path, filename)) print(image) # 新建一个图层,用于绘制水印 layer = Image.new("RGBA", image.size, (0,0,0,0)) draw = ImageDraw.Draw(layer) # 计算水印文字的位置 text_width, text_height = draw.textsize(text, font) x = image.width - text_width - 10 # x坐标 y = image.height - text_height - 10 # y坐标 # 绘制水印文字 draw.text((x,y), text, font=font, fill=(255,255,255,128)) # 合并图层 result = Image.alpha_composite(image, layer) print(result) # 保存处理后的图片 result.save(os.path.join(path, filename)) 优化这段代码 当前报错内容 File "E:\python-i\venv\lib\site-packages\PIL\Image.py", line 3168, in alpha_composite return im1._new(core.alpha_composite(im1.im, im2.im)) ValueError: image has wrong mode 图片是mode=RGB
时间: 2023-12-14 12:03:50 浏览: 37
根据你提供的代码和报错信息,可能是因为调用了 `alpha_composite()` 方法,该方法要求两张图片的模式必须相同。因此,你需要将新建的图层 `layer` 和打开的图片 `image` 都转换为 RGBA 模式才可以使用 `alpha_composite()` 方法。
以下是修改后的代码:
```python
from PIL import Image, ImageDraw, ImageFont
import os
text = "watermark" # 水印文字
font = ImageFont.truetype("arial.ttf", 36) # 字体和大小
path = r"C:\Users\Administrator\Desktop\0424特价直播样品\新建文件夹\新建文件夹"
for filename in os.listdir(path):
if filename.endswith(".jpg") or filename.endswith(".png") or filename.endswith(".JPG") or filename.endswith(".PNG"): # 只处理jpg和png图片
# 处理每一张图片
# 打开图片
image = Image.open(os.path.join(path, filename))
print(image.mode)
# 将图片转换为 RGBA 模式
if image.mode != 'RGBA':
image = image.convert('RGBA')
# 新建一个图层,用于绘制水印
layer = Image.new("RGBA", image.size, (0,0,0,0))
draw = ImageDraw.Draw(layer)
# 计算水印文字的位置
text_width, text_height = draw.textsize(text, font)
x = image.width - text_width - 10 # x坐标
y = image.height - text_height - 10 # y坐标
# 绘制水印文字
draw.text((x,y), text, font=font, fill=(255,255,255,128))
# 将图层转换为 RGBA 模式
if layer.mode != 'RGBA':
layer = layer.convert('RGBA')
# 合并图层
result = Image.alpha_composite(image, layer)
print(result.mode)
# 保存处理后的图片
result.save(os.path.join(path, filename))
```
在上述代码中,我们添加了两个 `if` 语句,用于检查图片和图层的模式是否为 RGBA 模式。如果不是,则使用 `convert()` 方法将其转换为 RGBA 模式。这样就可以避免 `alpha_composite()` 方法报错的问题。