diff --git a/dicom_filter.py b/dicom_filter.py index 3e25880..f7bc90b 100755 --- a/dicom_filter.py +++ b/dicom_filter.py @@ -8,8 +8,9 @@ import cv2 import json from pflog import pflog - -__version__ = '1.1.5' +from pydicom.pixel_data_handlers import convert_color_space +import numpy as np +__version__ = '1.1.6' DISPLAY_TITLE = r""" _ _ _ __ _ _ _ @@ -103,7 +104,8 @@ def save_as_image(dcm_file, output_file_path, file_ext): print(f"Saving output file as {output_file_path}") # Prevents color inversion happening while saving as images - cv2.imwrite(output_file_path, cv2.cvtColor(pixel_array_numpy,cv2.COLOR_RGB2BGR)) + rgb = convert_color_space(pixel_array_numpy, "YBR_FULL", "RGB") + cv2.imwrite(output_file_path,rgb) diff --git a/requirements.txt b/requirements.txt index baf5714..84c1189 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,9 +1,9 @@ chris_plugin==0.2.1 opencv-python -pydicom==2.1.2 +pydicom pflog==1.2.26 pftel-client~=1.0.6 # for bug fix on transfer syntax errors pylibjpeg pylibjpeg-libjpeg -python-gdcm \ No newline at end of file +python-gdcm