Grant Brewster

A potential workaround I found on this stack overflow (https://stackoverflow.com/questions/60507120/how-to-correctly-use-the-tensorflow-meaniou-metric): class MyMeanIOU(tf.keras.metrics.MeanIoU): def update_state(self, y_true, y_pred, sample_weight=None): return super().update_state(y_true, tf.argmax(y_pred, axis=-1), sample_weight) In this case my y_true was a mask...