PR3Wolf-Abgabe1/main.py

60 lines
1.8 KiB
Python

import SimpleITK as sitk
import numpy as np
import matplotlib.pyplot as plt
img = sitk.ReadImage("Thorax1.0B40f_Downsampled_by_2.nrrd")
xSp, ySp, zSp = img.GetSpacing()
volumePerVoxel = xSp * ySp * zSp
#print(f"Spacing: \n{xSp = }\n{ySp = }\n{zSp = }\n")
print(f"1) Die Voxel sind Quader mit den verschiedenen Seitenlängen: X: {xSp}, Y: {ySp}, Z: {zSp} \n")
sizeVector = img.GetSize()
xSi, ySi, zSi = sizeVector
print(f"Size: \n{xSi = }\n{ySi = }\n{zSi = }\n")
print(f"2) Das Bild hat in den drei Raumrichtungen {xSi * ySi * zSi} Voxel. \n")
print(f"3) Die Methode GetSize() gibt den Typ tuple zurück. \n")
#Sollte man kontrollieren
print(f"4) Der Bildausschnitt in die drei Raumrichtungen ist X: {xSp*xSi} mm, Y: {ySp*xSi} mm, Z: {zSp*xSi} mm\n")
# CT-Bild der Lunge erstellen
seedList = [(120,140,100)]
lungs = sitk.ConnectedThreshold(img, seedList , -1100, -200)
sitk.WriteImage(lungs, "lungs.nrrd")
# Volumen der Lunge berechnen
lungsArray = np.array(sitk.GetArrayFromImage(lungs))
voxelInLungs = lungsArray[lungsArray > 0].size * volumePerVoxel
lungsInLitre = voxelInLungs / 1000000
#print(lungsInLitre)
# Histogramm des CT-Bilds
thoraxArray = np.array(sitk.GetArrayFromImage(img))
thoraxArray1D = thoraxArray.flatten()
plt.hist(thoraxArray1D, bins=100)
plt.title("Helligkeitswerte eines Thorax")
plt.xlabel("Helligkeitswert in Hounsfield-Einheiten (HU)")
plt.ylabel("Häufigkeit in Millionen (mio)")
plt.show()
plt.clf()
# Alle Helligkeitswerte der Lunge seperieren
lungBrightness = []
lungsArray1D = lungsArray.flatten()
for (ind, elem) in enumerate(lungsArray1D):
if elem > 0:
lungBrightness.append(thoraxArray1D[ind])
# Histogram aller Helligkeitswerte in der Lunge
plt.hist(lungBrightness, bins=100)
plt.title("Helligkeitswerte einer Lunge")
plt.xlabel("Helligkeitswert in Hounsfield-Einheiten (HU)")
plt.ylabel("Häufigkeit")
plt.show()
plt.clf()