Я реализовал сеть FCN для семантической сегментации.Я использую Городские пейзажи в качестве моего набора данных.Как вы знаете, есть некоторые классы в Городских пейзажах, которые вы игнорируете во время обучения, и они обозначены как 255. Я использовал взвешенную потерю, чтобы игнорировать потерю для неизвестных классов (установите потерю равной нулю для неизвестного класса).Теперь я хочу исключить неизвестный класс из моей метрики оценки (среднее пересечение по объединению (mIOU)). Мне не ясно, как исключить неизвестный класс в этой точке.
В данный момент я рассматриваю всеклассы, включающие неизвестный класс, как этот, используя метод tenorflow:
miou, confusion_mat = tf.metrics.mean_iou(labels=annotation, predictions=pred_annotation, num_classes=num_cls)
with tf.control_dependencies([tf.identity(confusion_mat)]):
miou = tf.identity(miou)
Я пробовал это, но выдает ошибку для несвязанной метки (для метки unkonwn)
miou, confusion_mat = tf.metrics.mean_iou(labels=annotation, predictions=pred_annotation, num_classes=(num_cls-1))