handle color conversion while saving

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)
 
 
 

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
+python-gdcm