如何在pycharm软件中利用代码在文件夹A中的若干图片中添加6个值为0且大小为5乘5的小方格掩码(该掩码有规律的从上到下从左到右均匀居中分布于图片中)并导出到文件夹B
时间: 2024-05-03 22:20:54 浏览: 85
我可以为您解答关于在Pycharm中添加掩码的问题。
您可以使用Python中的Pillow库来实现这个功能。您可以使用代码打开文件夹A中的图片,添加掩码并导出到文件夹B。
下面是一份示例代码:
```
from PIL import Image
import os
# 设置掩码的大小和属性
mask_width = 5
mask_height = 5
mask_color = 0
# 遍历文件夹A中的所有图片
for filename in os.listdir("/path/to/folder/A"):
if filename.endswith(".jpg") or filename.endswith(".png"):
# 打开图片
img = Image.open(os.path.join("/path/to/folder/A", filename)).convert("RGBA")
# 计算掩码的位置
box_x = (img.width - mask_width * 3) // 2
box_y = (img.height - mask_height * 2) // 2
# 打开掩码
mask = Image.new("RGBA", (mask_width, mask_height), (mask_color, mask_color, mask_color, 255))
# 在图片上添加掩码
for i in range(6):
pos_x = box_x + (mask_width * (i % 3))
pos_y = box_y + (mask_height * (i // 3))
img.paste(mask, (pos_x, pos_y), mask)
# 导出图片到文件夹B
img.save(os.path.join("/path/to/folder/B", filename))
```
请注意将代码中的“/path/to/folder/A”和“/path/to/folder/B”替换成实际的文件夹路径。
希望这可以帮到您。
阅读全文