How to ues gpu to accelerate test image with classification.cpp

[任佑顏]

Hello everyone.

I have some question!

May I have anyone to solution my question.

I complier classification.cpp.

And I run with
http://caffe.berkeleyvision.org/gathered/examples/cpp_classification.html

I put the clock_t on the input image

Like this:

  clock_t clockbegin,clockend;

  clockbegin = clock();

  cv::Mat img = cv::imread(file, -1);

  //CHECK(!img.empty()) << "Unable to decode image " << file;

  std::vector<Prediction> predictions = classifier.Classify(img);

  clockend = clock();

  printf("%Lf\n", (long double)(clockend - clockbegin));

And the time is about 0.3s.

But it's maybe not use gpu.

How to ues gpu to accelerate this program classification.cpp.

And the time will less than 0.05s.

Thank everyone for help.

Have a nice day.

[Hossein Hasanpour]

Simply write :

 Caffe::set_mode(Caffe::GPU);

inside Classifier's constructor and you are good to go. 

[任佑顏]

To Hossein Hasanpour:

Thank you very much!!

I think it's too late to classify a image.

First time to classify a image,I think it almost 0.01s.

But it cost about 0.4s.

So I think it isn't a "GPU MODE"

But it looks already a GPU MODE.

Thank you so much 

Have a nice day