TensorFlow版本：在2.4 和 2.5上这样改就可以（已测试）
还有其他版本好像是调用 call 方法实现的IoU,所以需要对应需要修改 call 函数
Tensorflow中tf.keras.metrics.MeanIoU在预测返回值为one-hot编码的情况下使用IoU
class MeanIoU(tf.keras.metrics.MeanIoU):
def update_state(self, y_true, y_pred, sample_weight=NoneTensorflow中tf.keras.metrics.MeanIoU在shape不一致错误
当TensorFlow版本为2.4和2.5时，针对tf.keras.metrics.MeanIoU在计算IoU时出现的shape不一致错误，可以通过修改update_state函数处理one-hot编码的预测结果。网络输出的预测值是one-hot编码形式，而Label数据集使用了label-encode，这种情况下需要转换y_pred的格式以避免错误。
最低0.47元/天 解锁文章
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
