Сегментация изображения при помощи нейронной сети

Необходимо выполнить задание и прислать мне в формате *.ipynb-файл, в этом же файле нужно расписать, что делается.

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

Работа на сервере:

!nvidia-smi

os.environ['CUDA_VISIBLE_DEVICES'] = '-1'

Набор данных:

TRAIN_PATH = '/zfs/hybrilit.jinr.ru/scratch/SOGU-MAY2021/cell_count/data-science-bowl-2018/stage1_train/'

TRAINMASKS_PATH = '/zfs/hybrilit.jinr.ru/scratch/SOGU-MAY2021/cell_count/data-science-bowl-2018/stage1_train_combinedmasks/'

TEST_PATH = '/zfs/hybrilit.jinr.ru/scratch/SOGU-MAY2021/cell_count/data-science-bowl-2018/stage1_test/'