Как найти IoU из масок сегментации?

Question

Я выполняю задачу сегментации изображения и использую набор данных, который имеет только истинные факты, но не имеет ограничивающих рамок или многоугольников.

У меня есть 2 класса (игнорируя 0 для фона), а выходные данные и метки истинности заземления находятся в массиве, подобном

Предсказано - / --- Метки

0|0|0|1|2 0|0|0|1|2 0|2|1|0|0 0|2|1|0|0 0|0|1|1|1 0|0|1|1|1 0|0|0|0|1 0|0|0|0|1

Как рассчитать IoU из них?

PS: я использую python3 с pytorch api

Answer 1

Итак, я только что узнал, что jaccard_simility_score считается IoU.

Так что решение очень простое,

from sklearn.metrics import jaccard_similarity_score jac = jaccard_similarity_score(predictions, label, Normalize = True/False)

Ссылка на источник: http://scikit -learn.org / stable / modules / Генерированный / sklearn.metrics.jaccard_sdentifity_score.html