Оказание массива с поплавком для 24-разрядного изображения RGB (например, с использованием PIL)

Question

x - массив numpy.float32 со значениями от до 0. Это дБ (децибел).

Когда я делаю (в соответствии с рекомендациями here):

Image.fromarray(x, mode='F') 

я получаю в оттенках серого или иногда почти черное изображение.

Как сопоставить float в [-200, 0] с 24-битным массивом байтов RGB (с использованием цветовой схемы), который может быть прочитан с помощью модуля Python PIL с Image.fromarray(x, mode='RGB')?

---

Edit:

Необходимый .wav аудиофайл here, для которого мы хотим построить spectrogram.

Вот код для проверки:

import scipy, numpy as np 
import scipy.io.wavfile as wavfile 
import numpy as np 
from PIL import Image 

def stft(x, fftsize=1024, overlap=4): 
    hop = fftsize/overlap 
    w = scipy.hanning(fftsize+1)[:-1] 
    return np.array([np.fft.rfft(w*x[i:i+fftsize]) for i in range(0, len(x)-fftsize, hop)]) 

def dB(ratio): 
    return 20 * np.log10(ratio+1e-10) 

def magnitudedB(frame, fftsize=1024): 
    w = scipy.hanning(fftsize+1)[:-1] 
    ref = np.sum(w)/2 
    return dB(np.abs(frame)/ref) 

sr, x = wavfile.read('test.wav') 

x = np.float32(x)/2**15 

s = magnitudedB(stft(x)).astype(np.float32).transpose()[::-1,] 
print "Max %.1f dB, Min %.1f dB" % (np.max(s), np.min(s)) 

im = Image.fromarray(s+200, mode='F') 
im.show() 

Примечания:

• палитру является оттенки серого, как получить другую цветовую палитру? как this one

• Мое единственное требование заключается в том, что выходное изображение может быть прочитан в Tkinter кадр/холст (он хорошо работает с PIL-х im = Image.fromarray(...) затем ImageTk.PhotoImage(image=im)) или WxPython кадр/холст.

Answer 1

Based в ответе here вы можете использовать matplotlib colormaps для преобразования массива numpy перед преобразованием в изображение.

#im = Image.fromarray(s+200, mode='F') 
from matplotlib import cm 
s = (s + 200)/200.0 # input data should range from 0-1 
im = Image.fromarray(cm.jet(s, bytes=True)) 
im.show() 

Возможно, вы должны установить масштабирование соответствующим образом на основе ваших значений min/max. Выход

Пример:

Answer 2

Я не могу найти подробную информацию о режиме = «F» в документации, но я бы ожидать, что она принимает значения пикселов в диапазоне, как 0,0 - 1,0. Ваши значения полностью ниже этого диапазона, таким образом, черное изображение; вам нужно будет их преобразовать.

Получение изображения с комбинированной обработкой (вместо оттенков серого) потребует mode = 'P', что потребует преобразования данных в массив байтов.

Answer 3

Чтобы сфотографировать изображения с помощью цветовых карт, я предлагаю вам использовать matplotlib.pyplot.imshow.

Результат делать это с файлом test.wav будет что-то вроде этого:

Для получения более подробной информации о создании аудио спектрограммы с помощью питона вы можете прочитать больше о нем here