Updated the app to resolve PIL Image issue running in MAP container

Signed-off-by: M Q <mingmelvinq@nvidia.com>

Author: MMelQin

examples/apps/simple_imaging_app/gaussian_operator.py

@@ -58,6 +58,7 @@ def setup(self, spec: OperatorSpec):
         spec.param("channel_axis", 2)
 
     def compute(self, op_input, op_output, context):
+        import numpy as np
         from skimage.filters import gaussian
         from skimage.io import imsave
 
@@ -67,6 +68,13 @@ def compute(self, op_input, op_output, context):
         data_in = op_input.receive("in1")
         data_out = gaussian(data_in, sigma=self.sigma_default, channel_axis=self.channel_axis)
 
+        # Make sure the data type is what PIL Image can support, as the imsave function calls PIL Image fromarray()
+        # Some details can be found at https://stackoverflow.com/questions/55319949/pil-typeerror-cannot-handle-this-data-type
+        print(f"Data type of output: {type(data_out)!r}, max = {np.max(data_out)!r}")
+        if np.max(data_out) <= 1:
+            data_out = (data_out*255).astype(np.uint8)
+        print(f"Data type of output post conversion: {type(data_out)!r}, max = {np.max(data_out)!r}")
+
         # For now, use attribute of self to find the output path.
         self.output_folder.mkdir(parents=True, exist_ok=True)
         output_path = self.output_folder / "final_output.png"

notebooks/tutorials/01_simple_app.ipynb

@@ -3,7 +3,7 @@ pydicom>=2.3.0
 PyPDF2>=2.11.1
 highdicom>=0.18.2
 SimpleITK>=2.0.0
-Pillow>=8.0.0
+Pillow>=8.4.0
 numpy-stl>=2.12.0
 trimesh>=3.8.11
 nibabel>=3.2.1