Hiển thị mảng nổi cho hình ảnh RGB 24 bit (sử dụng PIL chẳng hạn)

Question

x là một mảng numpy.float32, với các giá trị từ -200 đến 0. Đây là các giá trị dB (decibel).

Khi tôi làm (theo khuyến cáo here):

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

tôi nhận được một tông màu xám hoặc hình ảnh đôi khi gần như màu đen.

Cách ánh xạ phao trong [-200, 0] tới mảng byte 24 bit (sử dụng bản đồ màu) có thể đọc được với mô-đun Python PIL với Image.fromarray(x, mode='RGB')?

---

Edit:

Các yêu cầu tập tin âm thanh .wav là here, mà chúng muốn âm mưu spectrogram.

Dưới đây là một số mã để kiểm tra:

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() 

Ghi chú:

• Các bản đồ màu là xám, làm thế nào để có được bản đồ màu khác? như this one

• Yêu cầu duy nhất của tôi là hình ảnh đầu ra có thể được đọc vào một khung Tkinter/canvas (nó hoạt động tốt với PIL của im = Image.fromarray(...) sau đó ImageTk.PhotoImage(image=im)) hoặc wxPython khung hình/canvas.

Answer 1

Dựa theo câu trả lời here, bạn có thể sử dụng matplotlib colormaps để chuyển mảng numpy trước khi chuyển đổi thành hình ảnh.

#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() 

Có lẽ bạn nên đặt tỷ lệ phù hợp dựa trên giá trị tối thiểu/tối đa của mình. sản lượng

mẫu:

Answer 2

tôi không thể tìm thấy bất kỳ chi tiết về chế độ = 'F' trong tài liệu, nhưng tôi mong chờ nó lấy giá trị pixel trong một phạm vi như 0,0-1,0. Giá trị của bạn hoàn toàn nằm dưới phạm vi đó, do đó hình ảnh màu đen; bạn sẽ cần phải biến đổi chúng.

Bắt hình ảnh có màu xám (thay vì thang độ xám) sẽ yêu cầu chế độ = 'P', điều này sẽ yêu cầu bạn chuyển dữ liệu của mình thành một mảng byte.

Answer 3

Để vẽ hình ảnh bằng colormaps, tôi khuyên bạn nên sử dụng matplotlib.pyplot.imshow.

Kết quả của việc làm như vậy với tập tin test.wav của bạn sẽ là một cái gì đó như thế này:

Để biết thông tin chi tiết hơn về việc tạo spectrograms âm thanh sử dụng python bạn có thể đọc thêm về nó here