Extract DICOMDIR slices and convert to mp4 file

Question

I have a DICOMDIR file with a lot of dcm files in it. I have extracted the .dcm file with Instance Number 1. It has 49 slices/frames. I wnat to convert those frames into an mp4 format. My code gives and output but the video is still and has a yellow hue overlayed over the video. How can I resolve this.

import numpy as np
import pydicom
import os
import cv2

dicomdir_path = 'dcm_files/DICOMDIR'
dicomdir = pydicom.dcmread(dicomdir_path)

record = None
for record in dicomdir.DirectoryRecordSequence:
    if record.DirectoryRecordType =='IMAGE' and record.InstanceNumber == 1:
        break

filename = os.path.join(os.path.dirname('dcm_files/DICOMDIR'), record.ReferencedFileID[1])
ds = pydicom.dcmread(filename)

num_frames = ds.NumberOfFrames

pixel_data = ds.pixel_array

if len(pixel_data.shape) > 2:
    pixel_data = pixel_data[0]

pixel_data = cv2.normalize(pixel_data, None, 0, 255, cv2.NORM_MINMAX, cv2.CV_8U)
print(pixel_data.shape)

num_rows, num_cols, _ = pixel_data.shape

pixel_data = pixel_data.reshape((num_rows, num_cols, 3))
pixel_data = [pixel_data] * num_frames
pixel_data = np.array(pixel_data)

print(pixel_data.shape)

fourcc = cv2.VideoWriter_fourcc(*"mp4v")
fps = ds.CineRate
video_writer = cv2.VideoWriter('Output/output.mp4', fourcc, fps, (num_cols, num_rows))

for i in range(num_frames):
    frame = pixel_data[i]
    if len(frame.shape) == 2:
        frame_bgr = cv2.cvtColor(frame, cv2.COLOR_GRAY2RGB)
    else:
        frame_bgr = frame
    video_writer.write(frame)

video_writer.release()

Answer 1

Finally got it. OpenCV has the option to convert dicom files with Photometric Interpretation YBR_FULL_422 to RGB using cv2.COLOR_YUV2RGB. Below is the finalised code

from glob import glob
import pydicom
import os
import cv2
from PIL import Image
import re
import numpy as np

# Load the DICOMDIR file
dicomdir_path = 'dcm_files/DICOMDIR'
dicomdir = pydicom.dcmread(dicomdir_path)

record = None
for record in dicomdir.DirectoryRecordSequence:
    if record.DirectoryRecordType =='IMAGE' and record.InstanceNumber == 1:
        break

filename = os.path.join(os.path.dirname('dcm_files/DICOMDIR'), record.ReferencedFileID[1])
ds = pydicom.dcmread(filename)
# print(ds)
num_frames = ds.NumberOfFrames

pixel_data = ds.pixel_array
pixel_data = cv2.normalize(pixel_data, None, 0, 255, cv2.NORM_MINMAX,cv2.CV_8U)
print(pixel_data.shape)

numbers = re.compile(r'(\d+)')
def numericalSort(value):
    parts = numbers.split(value)
    parts[1::2] = map(int, parts[1::2])
    return parts

# extract images from the dcm
for i in range(pixel_data.shape[0]):
    os.makedirs('Output/dicom_3d_images', exist_ok=True)
    slice = pixel_data[i, :, :, :]
    if ds.PhotometricInterpretation == 'YBR_FULL_422':
        slice = cv2.cvtColor(slice, cv2.COLOR_YUV2RGB)
    img = Image.fromarray(slice)
    img.save(f'Output/dicom_3d_images/image{i}.jpg')

# write images into a video
file_path = glob('Output/dicom_3d_images/*.jpg')
height, width, _ = cv2.imread(file_path[0]).shape
size = (width, height)
fourcc = cv2.VideoWriter_fourcc(*'mp4v')
fps = ds.CineRate
out = cv2.VideoWriter('Output/image1.mp4', fourcc, fps, size)

for filename in sorted(file_path, key=numericalSort):
    img = cv2.imread(filename)
    out.write(img)
out.release()