sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-12-22 14:15:01 +00:00
Code Issues Projects Releases Packages Wiki Activity

fixup

Browse source
This commit is contained in:
Starbeamrainbowlabs 2023-03-03 22:45:34 +00:00
parent 0734201107
commit c909cfd3d1
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
aimodel/src/lib/ai/components/MetricMeanIoU.py
View file

@ -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.