Определение опухоли мозга по снимкам МРТ

Описание

Создать нейронную сеть. Исрользуя Computer Vision, bounding boxes обучить модель модель находить на снимках области с раковой опухолью.

Решение

Используя Pytorch, Pillow, Sklearn была создана и обучена модель нейронной сети со сверточными слоями. Для увеличения количества входных даных, изображения были аугментированы.

Результат

Модель на тестовом наборе показала индекс Жаккарда 0.78.

Ссылки на примеры реализации

 github.com/SweetyKekS/Brain-tumor/blob/0a5a81689fd...

Презентация проекта