Precise multiple_colors.png file name

This commit is contained in:
Benjamin Loison 2024-04-23 05:01:22 +02:00
parent be9d4f8f4b
commit c65b3642d8
No known key found for this signature in database

View File

@ -4,8 +4,10 @@ from tqdm import tqdm
import matplotlib.pyplot as plt
import numpy as np
PREFIX = 'mean_rafael_arw_sky_mean_'
def getImageByColor(color):
filePath = f'means/mean_rafael_arw_bilateral_{color}.npy'
filePath = PREFIX + f'{color}.npy'
image = np.load(filePath)
return image
@ -39,4 +41,4 @@ for color in tqdm(Color, 'Color'):
multipleColorsImage[newX, newY] = pixel
plt.imsave('multiple_colors.png', multipleColorsImage)
plt.imsave(PREFIX + 'multiple_colors.png', multipleColorsImage)