Update nnunet/dataset_conversion/Task500_Glacier_reverse.py

nnunet/dataset_conversion/Task500_Glacier_reverse.py

@@ -39,19 +39,19 @@ def main(input_folder):
             color_zone[glacier_dil == 1] = 127
 
             color_front = extract_front_from_zones(color_zone, 10)
-            color_front[color_front==255] =1
+            color_front[color_front==255] = 255
 
             output_path_zone = join(output_folder, file[:-len('.nii.gz')] + '_zone.png')
             output_path_front = join(output_folder, file[:-len('.nii.gz')] + '_front.png')
 
-            imageio.imwrite(output_path_zone, color_zone)
-            imageio.imwrite(output_path_front, color_front)
-            """
-            img_zone = Image.fromarray(color_zone)
-            img_front = Image.fromarray(color_front)
+            #imageio.imwrite(output_path_zone, color_zone)
+            #imageio.imwrite(output_path_front, color_front)
+
+            img_zone = Image.fromarray(color_zone).convert('RGB')
+            img_front = Image.fromarray(color_front).convert('RGB')
             img_zone.save(output_path_zone)
             img_front.save(output_path_front)
-            """
+
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()