dicom2png.py

import cv2
import os
import pydicom

inputdir = './train/'
outdir = './png/'
#os.mkdir(outdir)

test_list = [ f for f in  os.listdir(inputdir)]

for f in test_list[:10]:   # remove "[:10]" to convert all images
    ds = pydicom.read_file(inputdir + f) # read dicom image
    img = ds.pixel_array # get image array
    cv2.imwrite(outdir + f.replace('.dicom','.png'),img) # write png image

import cv2
import os
import pydicom

inputdir = './train/'
outdir = './png/'
#os.mkdir(outdir)

test_list = [ f for f in  os.listdir(inputdir)]

for f in test_list[:10]:   # remove "[:10]" to convert all images
    ds = pydicom.read_file(inputdir + f) # read dicom image
    img = ds.pixel_array # get image array
    img = img.astype("float32")
    img = (img - img.max())/(img.max() - img.min())*255.
    cv2.imwrite(outdir + f.replace('.dicom','.png'),img) # write png image