如何在pycharm软件中利用代码在文件夹A中的若干图片中添加6个值为0且大小为5乘5的小方格掩码(均匀分布于图片中)并导出到文件夹B
时间: 2024-05-02 14:19:22 浏览: 149
这是一个关于图片处理的技术问题,您可以使用Pillow库来实现。代码示例如下:
```
from PIL import Image, ImageDraw
import os
# 图片路径
input_path = './A'
output_path = './B'
# 循环遍历文件夹内的所有图片
for file_name in os.listdir(input_path):
if file_name.endswith('.jpg') or file_name.endswith('.png'):
# 打开图片
image = Image.open(os.path.join(input_path, file_name))
# 获取图片宽度和高度
width, height = image.size
# 创建一个大小为5x5的小方格掩码
mask = Image.new('L', (5,5), 0)
# 在图片上随机位置添加6个小方格掩码
for i in range(6):
x = random.randint(0, width-5)
y = random.randint(0, height-5)
image.paste(mask, (x,y))
# 保存处理后的图片到输出文件夹
output_file_name = os.path.join(output_path, file_name)
image.save(output_file_name)
```
以上代码使用Pillow库打开图片,随机添加了6个小方格掩码。您可以根据需要修改代码中的路径和掩码的大小进行调整。
阅读全文