写一段读取文件夹中的所有图像,同时使用小波变换对文件夹中的所有图像进行增强,然后将增强后的所有图像保存到另一个文件夹的python代码
时间: 2024-03-14 14:47:35 浏览: 51
Python 读取指定文件夹下的所有图像方法
好的,以下是Python代码示例:
```python
import os
import cv2
import pywt
import numpy as np
# 定义小波变换函数
def wavelet_transform(img):
coeffs2 = pywt.dwt2(img, 'haar')
LL, (LH, HL, HH) = coeffs2
LL = cv2.normalize(LL, None, 0, 255, cv2.NORM_MINMAX)
return LL.astype(np.uint8)
# 读取文件夹中的所有图像
img_dir = './input_folder'
output_dir = './output_folder'
if not os.path.exists(output_dir):
os.makedirs(output_dir)
for filename in os.listdir(img_dir):
if filename.endswith('.jpg') or filename.endswith('.png'):
# 读取图像
img_path = os.path.join(img_dir, filename)
img = cv2.imread(img_path, 0)
# 小波变换增强
enhanced_img = wavelet_transform(img)
# 保存增强后的图像
output_path = os.path.join(output_dir, filename)
cv2.imwrite(output_path, enhanced_img)
```
这段代码会读取 `input_folder` 文件夹中的所有图像,进行小波变换增强后,将增强后的图像保存到 `output_folder` 文件夹中。你可以将代码中的文件夹路径改为你自己的路径,并根据需要修改小波变换的类型和参数。
阅读全文