python-scripts/scripts/rgb_to_grayscale/script.py

12 lines
366 B
Python
Raw Normal View History

2022-10-04 10:34:34 +00:00
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.image as mpimg
def rgb2gray(png_file='file_example.png'):
img = mpimg.imread(png_file)
gray = np.dot(img[:,:,:3], [0.2989, 0.5870, 0.1140])
plt.imshow(gray, cmap=plt.get_cmap('gray'), vmin=0, vmax=1)
plt.axis('off')
plt.savefig('gray_' + png_file + '.png', dpi=300)