Преобразовать сюжет в nparray

Question

У меня есть массив цифр (74, 743), который представляет спектрограмму человеческой речи в течение нескольких секунд.Я могу легко преобразовать это в график matplotlib, используя plt.subplots.matshow, но я хочу знать, возможно ли преобразовать график в исходный массив numpy?Как минимум, как matplotlib генерирует изображение из массива произвольной формы?

Я пытаюсь создать генерирующую Adverserial Network, которая будет генерировать изображения (это связано с превосходной производительностью сети при генерации изображений) спектрограмм,Затем я хочу преобразовать эти изображения спектрограмм в количественные спектрограммы, то есть построить график в виде массива.

Answer 1

Кажется, вы хотите применить цветовую карту к двумерному массиву. При использовании инструментов matplotlib это может выглядеть как

import numpy as np
from matplotlib.colors import Normalize
import matplotlib.cm as cm

data = np.random.rand(74, 743)

cmap = cm.viridis
norm = Normalize(data.min(), data.max())

output = cmap(norm(data))

print(output.shape)

Вывод представляет собой массив формы (74, 743, 4) со значениями от 0 до 1, обозначающий цвета RGBA.