mnist example

// Set device id and mode

Caffe::SetDevice(FLAGS_gpu); // Device ID

Caffe::set_mode(Caffe::GPU); // Set up Caffe: CPU or GPU

Caffe::set_phase(Caffe::TEST); // Phase설정: TRAIN or TEST

// Load net

Net<float> net("./model/mnist/lenet_deploy.prototxt");

net.CopyTrainedLayersFrom("./data/mnist/snapshot/lenet_iter_10000.caffemodel");

vector<Blob<float>*> input_vec;

Blob<float> blob(1, 1, IMAGE_MNIST_WIDTH, IMAGE_MNIST_WIDTH);

// copy image data (small_image) to blob

int i, j;

for (i = 0; i<IMAGE_MNIST_WIDTH; i++) // height

{

for (j = 0; j< IMAGE_MNIST_WIDTH; j++) // width

{

blob.mutable_cpu_data()[blob.offset(0, 0, i, j)] = (float)(unsigned                         char)small_image.at<uchar>(i, j); // gray

}

}

// set the patch for testing

input_vec.push_back(&blob);

float loss = 0.0;

const vector<Blob<float>*> & results = net.Forward(input_vec, &loss);

float *output = results[0]->mutable_cpu_data(); // 10