Skip to content

Instantly share code, notes, and snippets.

View ifadams's full-sized avatar

Ian ifadams

  • Intel Labs
  • Portland Metro Area
View GitHub Profile
@erogol
erogol / CaffeBatchPrediction.cpp
Created July 13, 2015 12:54
Caffe c++ batch based prediction
#include "caffeclassifier.h"
CaffeClassifier::CaffeClassifier(const string& model_file,
const string& trained_file,
const string& mean_file,
const string& label_file,
const bool use_GPU,
const int batch_size) {
if (use_GPU)
Caffe::set_mode(Caffe::GPU);