Я адаптировал обобщенную потерю игральных костей, предоставленную niftynet, для использования в собственной сети тензорного потока, которая выдает вероятность для одной метки. Чтобы использовать обобщенную потерю игральных костей, я добавил эту строку в начало файла (я работаю с 2D данными):
prediction = tf.concat([1 - prediction, prediction], axis=3)
Я использую размер пакета 1. Когда я тестировал type_weight для Square, Simple и Uniform, я получаю очень похожие результаты. Кажется, что ref_vol должен быть суммирован, или я пропустил это?