Add a model loader/simulator that actually loads caffe models.

This commit is contained in:
2017-10-07 17:33:11 +02:00
parent 4622c7c8c4
commit a034d2a7f3
5 changed files with 54 additions and 5 deletions

22
src/Simulator.hpp Normal file
View File

@@ -0,0 +1,22 @@
#pragma once
#include <string>
#include <memory>
#include <caffe/caffe.hpp>
namespace fmri {
using std::string;
class Simulator {
public:
typedef float DType;
Simulator(const string& model_file, const string& weights_file);
void simulate(const string& input_file);
private:
caffe::Net<DType> net;
};
}