Skip to content

GitLab

Commit 74ecea0d authored by Fan Yang's avatar Fan Yang Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 355946044
parent de14103c
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
official/vision/beta/evaluation/segmentation_metrics.py
View file @ 74ecea0d
......@@ -44,7 +44,7 @@ class MeanIoU(tf.keras.metrics.MeanIoU):
num_classes=num_classes, name=name, dtype=dtype)
def update_state(self, y_true, y_pred):
"""Updates metic state.
"""Updates metric state.
Args:
y_true: `dict`, dictionary with the following name, and key values.
......@@ -122,4 +122,3 @@ class MeanIoU(tf.keras.metrics.MeanIoU):
super(MeanIoU, self).update_state(
flatten_masks, flatten_predictions,
tf.cast(flatten_valid_masks, tf.float32))
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment