Fix(2d_segmentation): Restore the predicted value of seg image to the range of 0~225 before saving it

Author: Reqwey

2d_segmentation/torch/unet_evaluation_array.py

@@ -72,6 +72,7 @@ def main(tempdir):
             # compute metric for current iteration
             dice_metric(y_pred=val_outputs, y=val_labels)
             for val_output in val_outputs:
+                val_output = (val_output * 255).to(torch.uint8)
                 saver(val_output)
         # aggregate the final mean dice result
         print("evaluation metric:", dice_metric.aggregate().item())

2d_segmentation/torch/unet_evaluation_dict.py

@@ -87,6 +87,7 @@ def main(tempdir):
             # compute metric for current iteration
             dice_metric(y_pred=val_outputs, y=val_labels)
             for val_output in val_outputs:
+                val_output = (val_output * 255).to(torch.uint8)
                 saver(val_output)
         # aggregate the final mean dice result
         print("evaluation metric:", dice_metric.aggregate().item())