TensorFlowのU-NETで重みの保存と読み込みをしたい
TensorFlowのU-NETを使用したSemantic Segmentationを利用したいと思っています。
重みの保存と読み込みには、通常
w = tf.get_variable
b = tf.get_variable
saver = tf.train.Saver()
saver.restore(sess, "model.ckpt")
を用いると思います。
しかし、「U-NetでPascal VOC 2012の画像をSemantic Segmentationする (TensorFlow)」モデルでは、上記のvariableがありません。
恐らく、U=NET独自の手法があるものと思われます。
従いまして、重みの保存と読み込みをする方法がわかる方いましたら、ご教示頂けないでしょうか。
【サイト】
https://qiita.com/tktktks10/items/0f551aea27d2f62ef708
【ソース】
https://github.com/tks10/segmentation_unet
【参考(学習結果)】
Epoch: 248
[Train] Loss: 0.059211913 Accuracy: 0.9756693
[Test] Loss: 2.6319973 Accuracy: 0.7496191
Epoch: 249
[Train] Loss: 0.057428934 Accuracy: 0.9761983
[Test] Loss: 2.6868184 Accuracy: 0.74921834
Result
[Test] Loss: 2.6868184 Accuracy: 0.74921834
よろしくお願いします。