如何在pycharm软件中利用代码在文件夹A中的图片中添加6个值为0(均匀分布于图片中)且大小为5*5的小方格掩码并导出到文件夹B
时间: 2024-05-09 16:18:04 浏览: 97
您可以使用以下代码来实现您的需求:
```python
# 导入必要的库
import os
import cv2
import random
# 设置图片的路径
source_path = '文件夹A/'
target_path = '文件夹B/'
# 遍历文件夹中的所有图片
for file_name in os.listdir(source_path):
if file_name.endswith('.jpg') or file_name.endswith('.png'):
# 读取图片
image = cv2.imread(os.path.join(source_path, file_name))
# 在图片中添加掩码
for i in range(6):
x = random.randint(0, image.shape[1] - 5)
y = random.randint(0, image.shape[0] - 5)
image[y:y+5, x:x+5] = [0, 0, 0]
# 导出图片
cv2.imwrite(os.path.join(target_path, file_name), image)
```
这段代码将在文件夹A中遍历所有的图片文件,对于每一张图片,它会在其中随机添加6个5*5大小的小方格掩码,然后将处理后的图片导出到文件夹B中。注意,掩码是通过将随机的5*5大小的区域赋值为0来实现的,即将小方格填充为黑色。
阅读全文