import numpy as np
import matplotlib.pyplot as plt
from scipy.misc import imresize
from matplotlib import cm
# Преобразуем спектрограмму в log-шкалу и нормализуем значения
Zxx_log = np.log10(np.abs(Zxx))
Zxx_norm = (Zxx_log - Zxx_log.min()) / (Zxx_log.max() - Zxx_log.min())
# Используем colormap 'viridis' для преобразования значений в цвета
cmap = cm.get_cmap('viridis')
Zxx_rgb = cmap(Zxx_norm)
# Изменяем размер изображения до 128x128
Zxx_resized = imresize(Zxx_rgb, (128, 128))
# Переводим изображение в формат (3, 128, 128)
Zxx_final = np.transpose(Zxx_resized, (2, 0, 1))
# Выводим изображение
plt.imshow(Zxx_final)
plt.show()
