Задание по использованию цветового пространства

Используя архив, приложенный в файлах, нужно реализовать метод segment_fish в скрипте main.py. Результат работы будет оцениваться по метрике IoU между вашими масками и размеченными. Метод segment_fish получает изображение в формате BGR, и должен вернуть маску рыб (смотрите код).

Предлагается проверять алгоритм на обучающей выборке (dataset/train). Можно использовать файл main.ipynb для визуализации и проверки. При запуске скрипта main.py вы увидите IoU на обучающих данных.

Результатом должна быть метрика IoU > 0.55.