0

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)
Robert
  • 7,394
  • 40
  • 45
  • 64
Anki
  • 1

0 Answers0