Lenet Train Test

test-lenet-train-test.py

__author__ = 'caffedemo'

import caffe
import os
from PIL import Image

CAFFE_ROOT = "/home/caffedemo/caffe/caffe/"
TRAIN_TEST_NET = os.path.join(CAFFE_ROOT, 'examples/mnist/lenet_train_test.prototxt')


def ut_test_lenet_train_test():
    os.chdir(CAFFE_ROOT)
    net = caffe.Net(TRAIN_TEST_NET, caffe.TEST)
    result = net.forward()
    Image.fromarray(net.blobs['data'].data[0,:].reshape((28,28))*255).show()
    pass

if __name__ == '__main__':
    ut_test_lenet_train_test()