图片路径改成和输出路径相同
时间: 2024-04-10 22:28:13 浏览: 85
如果你想将去均值处理后的图片保存在与原始图片相同的目录中,可以将输出路径设置为与输入路径相同的目录。以下是修改后的代码示例:
```python
import os
import cv2
import numpy as np
def subtract_mean(image):
mean = np.mean(image)
return image - mean
# 假设图片路径存储在一个列表中,例如:image_paths = ['image1.jpg', 'image2.jpg', ...]
image_paths = ['image1.jpg', 'image2.jpg', ...]
for i, image_path in enumerate(image_paths):
# 读取图片
image = cv2.imread(image_path)
# 将图片转为浮点型
image = image.astype(float)
# 去均值处理
image = subtract_mean(image)
# 将图片转回无符号8位整型
image = np.clip(image, 0, 255).astype(np.uint8)
# 获取输入图片的目录路径
input_dir = os.path.dirname(image_path)
# 获取输入图片的文件名
filename = os.path.basename(image_path)
# 构建输出路径,保存在与输入图片相同的目录中
output_path = os.path.join(input_dir, f'result_{i+1}.jpg')
# 保存处理后的图片
cv2.imwrite(output_path, image)
print('图片去均值处理完成!')
```
这样修改后,处理后的图片将会保存在与输入图片相同的目录中。请确保 `image_paths` 列表中存储了正确的图片路径。
阅读全文