Skip to content

Added segmentation accuracy scripts

Valentina Kustikova requested to merge segmentation_accuracy-scripts into master

Created by: zhiltsov-max

Скрипты для оценки качества сегментации. На данный момент работают, но выдают результаты, расходящиеся со скриптом из оригинальной поставки deeplab (разница в 5-10% в сторону ухудшения оценок). Для работы скриптов требуется библиотека PIL (или Pillow):

$python -m pip install pillow

Скрипты написаны под python3 и не совместимы с python2.7.

Подход в первой версии: подсчет среднего IoU по классу среди всех изображений, содержащих данный класс Подход во второй версии: подсчет суммарного пересечения и объединения по каждому классу среди всех изображений

Использование:

$python segmentation_accuracy.py segmentation/dir gt/dir images_list.txt

Merge request reports