Python代码:将三个文件夹中的特征图进行相乘并保存到新文件夹
import os
from PIL import Image
folder1 = r'D:\论文代码\spectrograms'
folder2 = r'D:\论文代码\语谱图'
folder3 = r'D:\论文代码\MFCC'
new_folder = r'D:\论文代码\feature_multiplication'
for root, dirs, files in os.walk(folder1):
for file in files:
if file.endswith('.png'):
folder1_path = os.path.join(root, file)
folder2_path = folder1_path.replace(folder1, folder2)
folder3_path = folder1_path.replace(folder1, folder3)
new_path = folder1_path.replace(folder1, new_folder)
if not os.path.exists(os.path.dirname(new_path)):
os.makedirs(os.path.dirname(new_path))
img1 = Image.open(folder1_path)
img2 = Image.open(folder2_path)
img3 = Image.open(folder3_path)
img_new = Image.new('RGB', img1.size)
pixels_new = img_new.load()
pixels1 = img1.load()
pixels2 = img2.load()
pixels3 = img3.load()
for i in range(img1.size[0]):
for j in range(img1.size[1]):
r = pixels1[i, j][0]*pixels2[i, j][0]*pixels3[i, j][0] // 255
g = pixels1[i, j][1]*pixels2[i, j][1]*pixels3[i, j][1] // 255
b = pixels1[i, j][2]*pixels2[i, j][2]*pixels3[i, j][2] // 255
pixels_new[i, j] = (r, g, b)
img_new.save(new_path)
原文地址: http://www.cveoy.top/t/topic/nI8b 著作权归作者所有。请勿转载和采集!