mirror of
https://github.com/sbrl/research-rainfallradar
synced 2024-11-22 01:12: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
|
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)
|
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.
|
"""Compute the mean IoU for one-hot tensors.
|
||||||
Args:
|
Args:
|
||||||
y_true (tf.Tensor): The ground truth label.
|
y_true (tf.Tensor): The ground truth label.
|
||||||
|
|
Loading…
Reference in a new issue