мультиметочная сегментация в тензорном потоке 2.0 - PullRequest
0 голосов
/ 17 октября 2019

Я работаю над задачей сегментации, где каждый пиксель может принадлежать нескольким классам - сегментация по нескольким меткам. Основная правда и вывод (я получаю из U-Net) - None x H x W x 3 в форме. По сути, каждый пиксель представлен вектором размера 3. Правильно ли я поступаю, если вызываю compile на модели следующим образом:

model.compile(
    optimizer='rmsprop',
    loss='binary_crossentropy')

Или мне нужно написать собственную функцию потерь для этогозадание?

Спасибо тебе!

...