如何在pycharm软件中利用代码对文件夹A中的图片添加6个值为0的小方格掩码并导出到文件夹B
时间: 2024-05-14 09:12:58 浏览: 104
你可以使用Python中的OpenCV库来处理图像文件。下面是一个示例代码来添加6个小方格掩码并导出到文件夹B:
```python
import cv2
import os
# 定义图像文件夹A和目标文件夹B的路径
folderA = "path/to/folderA"
folderB = "path/to/folderB"
# 遍历文件夹A中的所有图像文件
for file in os.listdir(folderA):
# 如果文件是图像文件(JPEG、PNG等),则读取
if file.endswith(".jpg") or file.endswith(".png"):
# 读取图像
img = cv2.imread(os.path.join(folderA, file))
# 获取图像的尺寸
height, width, channels = img.shape
# 计算小方格的大小
square_size = int(min(height, width) / 6)
# 在图像上添加6个小方格
for i in range(6):
for j in range(6):
x = i * square_size
y = j * square_size
img[y:y+square_size, x:x+square_size] = [0, 0, 0]
# 保存图像到目标文件夹B
cv2.imwrite(os.path.join(folderB, file), img)
```
这个代码将遍历文件夹A中的所有图像文件,对于每个图像文件,它将读取图像并获取其尺寸。然后,它将计算每个小方格的大小,并在图像上添加6个小方格。最后,它将保存图像到目标文件夹B。
阅读全文