DICOM to PNG Converter

dicom_2_png.py

# coding: utf-8
__author__ = 'ZFTurbo: https://github.com/ZFTurbo/'

import os
import glob
import cv2
import pydicom
import argparse

def convert_dataset(args):
    inp_path = args.input_folder
    out_path = args.output_folder
    if not os.path.isdir(out_path):
        os.mkdir(out_path)

    files = glob.glob(inp_path + '*/**')
    print('Files: {}'.format(len(files)))

    for f in files:
        print(f)
        try:
            ds = pydicom.dcmread(f)
        except:
            print('cant read file!')
            continue
        img = ds.pixel_array
        print(img.shape, img.dtype)
        cv2.imwrite(out_path + os.path.basename(f) + '.png', img)


if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument("--input_folder", type=str, default='./', help="Folder where dicom files are located")
    parser.add_argument("--output_folder", type=str, default='./', help="Folder where to store PNG files")
    args = parser.parse_args()
    convert_dataset(args)