Add a model loader/simulator that actually loads caffe models.
This commit is contained in:
22
src/Simulator.hpp
Normal file
22
src/Simulator.hpp
Normal 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;
|
||||
};
|
||||
}
|
||||
Reference in New Issue
Block a user