naive_tf_test.py 文件源码

def main():
    model_filename = '../fcn8s-heavy-pascal.mat'
    input_image_filename = '../cat.jpg'

    caffe_mat = np.load(model_filename)
    image = build_image(input_image_filename)
    net = build_fcn8s(caffe_mat, image)
    feed_dict = {
        net['input']: image
    }

    with tf.Session() as sess:
        sess.run(tf.initialize_all_variables())
        result = sess.run(tf.argmax(net['score'], dimension=3),
                          feed_dict=feed_dict)

    save_image(result)