如何利用代码对文件夹A中的图片添加6个值为0的小方格掩码
时间: 2024-05-07 17:16:32 浏览: 11
可以使用Python的OpenCV库对图片进行处理。首先,使用os库遍历文件夹A中的所有图片,然后使用cv2.imread()函数读取图片。接着,使用cv2.copyMakeBorder()函数在图片周围添加6个值为0的小方格掩码,最后使用cv2.imwrite()将修改后的图片保存回原文件夹。以下是示例代码:
```
import cv2
import os
folder_path = 'A' # 文件夹名
for filename in os.listdir(folder_path):
if filename.endswith('.jpg') or filename.endswith('.png'): # 只处理图片文件
img_path = os.path.join(folder_path, filename)
img = cv2.imread(img_path)
top, bottom, left, right = 6, 6, 6, 6 # 添加的小方格掩码大小
img_with_border = cv2.copyMakeBorder(img, top, bottom, left, right,
cv2.BORDER_CONSTANT, value=[0, 0, 0])
cv2.imwrite(img_path, img_with_border) # 覆盖原文件
```
注意:在运行代码之前,需要先安装OpenCV库。可以使用以下命令在命令行中安装:
```
pip install opencv-python
```