diff --git a/aimodel/src/deeplabv3_plus_test_rainfall.py b/aimodel/src/deeplabv3_plus_test_rainfall.py index f73819b..666e14f 100755 --- a/aimodel/src/deeplabv3_plus_test_rainfall.py +++ b/aimodel/src/deeplabv3_plus_test_rainfall.py @@ -166,7 +166,7 @@ else: "metric_dice_coefficient": dice_coefficient, "make_sensitivity": sensitivity, "specificity": specificity, - "make_one_hot_mean_iou": mean_iou + "one_hot_mean_iou": mean_iou })