mirror of
https://github.com/sbrl/research-rainfallradar
synced 2024-11-25 02:32:59 +00:00
fixup
This commit is contained in:
parent
0734201107
commit
c909cfd3d1
1 changed files with 2 additions and 2 deletions
|
@ -3,9 +3,9 @@ import math
|
|||
import tensorflow as tf
|
||||
|
||||
|
||||
def make_one_hot_mean_iou():
|
||||
def make_one_hot_mean_iou(classes=2):
|
||||
iou = tf.keras.metrics.MeanIoU(num_classes=classes)
|
||||
def one_hot_mean_iou(y_true, y_pred, classes=2):
|
||||
def one_hot_mean_iou(y_true, y_pred, ):
|
||||
"""Compute the mean IoU for one-hot tensors.
|
||||
Args:
|
||||
y_true (tf.Tensor): The ground truth label.
|
||||
|
|
Loading…
Reference in a new issue