Correct issue convert to rtstruct with rt_utils

pyplati · Jul 26, 2021 · ca28822 · ca28822
1 parent 1f585ab
commit ca28822
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/.gitattributes b/.gitattributes
@@ -4,6 +4,9 @@
 
 $.dcm binary
 *.tar.gz binary
+*.png binary
+*.ttf binary
+*.woff binary
 
 *.ipynb filter=nbstripout
 *.ipynb diff=ipynb
diff --git a/platipy/dicom/io/nifti_to_rtstruct.py b/platipy/dicom/io/nifti_to_rtstruct.py
@@ -74,7 +74,7 @@ def convert_nifti(dcm_path, mask_input, output_file, color_map=cm.get_cmap("rain
             mask = sitk.ReadImage(str(mask))
 
         bool_arr = sitk.GetArrayFromImage(mask) != 0
-        bool_arr = np.transpose(bool_arr, (2, 1, 0))
+        bool_arr = np.transpose(bool_arr, (1, 2, 0))
         rtstruct.add_roi(mask=bool_arr, color=color, name=mask_name)
 
     rtstruct.save(str(output_file))