Сохранение моих предсказанных вероятностей на тестовом изображении для семантической сегментации, которые прогнозируются во время теста, в файл .mat в тензорном потоке - PullRequest
0 голосов
/ 28 сентября 2018

Итак, я хочу увидеть прогнозируемые значения вероятности на тестовом изображении, обученном в PSPNet для семантической сегментации.Я могу генерировать сегментированное выходное изображение.Однако я хочу увидеть, с какой уверенностью делается прогноз, и сохранить эти прогнозы в файле .mat, чтобы найти достоверность.Это мой код.

raw_output = net.layers['conv6'] # The output from my neural network.
raw_prediction = tf.reshape(raw_output_up, [-1, num_classes])
probs = tf.nn.softmax(raw_prediction, name='probs')
a = {}
a['values'] = probs
sio.savemat('probvalue.mat', a)

Однако, когда я просматриваю сгенерированный файл probvalue.mat, я вообще не вижу никаких значений.Так я тут что-то не так делаю?

Я также пытался сохранить значения logits непосредственно в файл .mat, но когда я пытаюсь открыть его, я получаю сообщение «Невозможно прочитать MAT-файл /probvalue.mat. Не двоичный файл MAT. Попробуйтезагрузить -ASCII для чтения в виде текста. "

Любая помощь будет принята с благодарностью.

Спасибо.

...