Unkomprimiertes Bild aus Byte-Array lesen

Question

Ich habe ein unkomprimiertes dicom-Video von einem Ultraschallgerät erstellt. Jetzt möchte ich es Frame für Frame in einer Python-Anwendung lesen und vorerst die Datei speichern. Später möchte ich etwas Bildverarbeitung hinzufügen. Bisher habe ich versucht, die zum ersten Frame gehörenden Bytes zu extrahieren.

import dicom 
import array 
from PIL import Image 

filename = 'image.dcm' 
img = dicom.read_file(filename) 
byte_array = array.array('B', img.PixelData[0:(img.Rows * img.Columns)]) 

Wie bekomme ich jetzt dieses Bytearray in eine Datei (Bitmap, JPEG was auch immer)? Ich habe versucht, die Python-Bild-Bibliothek mit image = Image.fromarray(byte_array), aber habe einen Fehler.

AttributeError: 'str' object has no attribute 'array_interface'

Ich denke, irgendwo muss ich auch die Abmessungen des Bildes angeben, aber habe nicht herausgefunden, wie.

Answer 1

Dank der Kommentare habe ich herausgefunden, wie man es löst. Das Bild war in 'RGB' und die Form des Arrays war (3L, 800L, 376L). Anstatt es in ein Byte-Array umzuwandeln, kann ich das pixel_array als ein numpiges Array nehmen und es zu (800L, 376L, 3L) umformen.

import dicom 
from PIL import Image 

filename = 'image.dcm' 
img = dicom.read_file(filename) 
output = img.pixel_array.reshape((img.Rows, img.Columns, 3)) 
image = Image.fromarray(output).convert('LA') 
image.save('output.png')