import pydicom,re
import numpy as np
import SimpleITK as sitk
import nibabel as nib
import os
base_image = 'E:/cardic_data/training/'
dicom_path = 'E:/cardic_dicom_data'
for dirpath, dirnames, files in os.walk(base_image):
for name in dirnames:
image_path = os.path.join(base_image, name)
path = os.path.join(dicom_path,name)
if not os.path.exists(path):
os.mkdir(path)
#print(image_path)
for patient in files:
match = re.search('patient(\d{03})',image_path)
patient_id = match.group(1)
img_name = 'patient{:s}_4d.nii.gz'.format(patient_id)
img_full_path = os.path.join(img_name,image_path)
print(img_full_path)
itk_img = sitk.ReadImage(img_full_path+'/patient{:s}_4d.nii.gz'.format(patient_id))
img = sitk.GetArrayFromImage(itk_img)
img = np.transpose(img, (1, 0, 2, 3))
print(img.shape)
for i in range(img.shape[0]):
for j in range(img.shape[1]):
out = sitk.GetImageFromArray(img[i,j,:,:].astype('int16'))
sitk.WriteImage(out,path+'/patient{}_4d_{}_{}.dcm'.format(patient_id,i, j))
#match = re.search('patient(\d{03})',base_image)
#patient_id = match.group(1)
#print(patient_id)
'''''
#itk_img = sitk.ReadImage('E:/cardic_data/training/patient003/patient003_4d.nii.gz')
#data_keys = os.path.join('E:/cardic_data/training/patient003','data_keys.txt')
#itk_img = sitk.ReadImage('E:/DICOM_data_cardic/patient001_4d_0_1.dcm')
img = sitk.GetArrayFromImage(itk_img)
img =np.transpose(img,(1,0,2,3))
keys = itk_img.GetMetaDataKeys()
#with open(data_keys, 'w') as f:
print(keys)
print(keys)
for key in keys:
print (key,':', itk_img.GetMetaData(key))
print("img shape:",img.shape)
for i in range(img.shape[0]):
for j in range(img.shape[1]):
out = sitk.GetImageFromArray(img[i,j,:,:])
#keys = out.LoadPrivateTagsOn()
#for key in keys:
# print(key, ':', itk_img.GetMetaData(key))
#print(keys)
sitk.WriteImage(out, 'E:/DICOM_data_cardic/patient003/patient003_4d_{}_{}.dcm'.format(i,j))
#print(img[i,j,:,:].shape)
'''''
评论0