Whenever we use the series description and image orientation patient tag both time it gives different pics list?
import pydicom
import os
def select_transverse_images(root_dir):
transverse_images = []
root_dir = 'D:\\prostate images\\Train\\ProstateX'
# iterate over subdirectories
for sub1_dir_name in os.listdir(root_dir):
sub1_dir_path = os.path.join(root_dir, sub1_dir_name)
if os.path.isdir(sub1_dir_path):
# iterate over sub-subdirectories
for sub2_dir_name in os.listdir(sub1_dir_path):
sub2_dir_path = os.path.join(sub1_dir_path, sub2_dir_name)
# iterate over sub-sub-subdirectories
for sub3_dir_name in os.listdir(sub2_dir_path):
sub3_dir_path = os.path.join(sub2_dir_path, sub3_dir_name)
# iterate over DICOM files
for file_name in os.listdir(sub3_dir_path):
if file_name.endswith(".dcm"):
file_path = os.path.join(sub3_dir_path, file_name)
ds = pydicom.dcmread(file_path)
#if ds.ImageOrientationPatient[:2] in ([1, 0], [-1, 0]):
if ds.ImageOrientationPatient[:6] in ([1, 0, 0, 0, 1, 0], [-1, 0, 0, 0, -1, 0]):
print(f"Found transverse image: {file_path}")
transverse_images.append(ds)
return transverse_images
dataset_dir = 'D:\\prostate images\\Train\\ProstateX'
transverse_images = select_transverse_images(dataset_dir)