代码中需要更改的参数有三个:
1.original_folder:原始文件夹路径
2.new_folder:保存新文件夹路径
3.img1(最大值法),img2(平均值法),img3(加权平均法)
import os
import cv2
import numpy as np
# 原始文件夹路径
original_folder = r"C:\Users\12302\Desktop\training-images\tulips/"
# 保存的新文件夹路径
new_folder = r'C:\Users\12302\Desktop\training-images\tulips'
# 遍历原始文件夹中的图像
for filename in os.listdir(original_folder):
img = cv2.imread(os.path.join(original_folder, filename), 1)
# img1=img.max(axis=2)
# img2=img.mean(axis=2)
img3 = np.dot(img, [0.299, 0.587, 0.114])
# cv2.imwrite(os.path.join(new_folder, '1'+filename), img1)
# cv2.imwrite(os.path.join(new_folder, '2'+filename), img2)
cv2.imwrite(os.path.join(new_folder, '3'+filename), img3)