Create an intermediate representation.

This way the visualiser does not need to know all about caffe, and can
just work on the intermediate representation which is a lot easier on
the compiler.
This commit is contained in:
2017-10-12 14:21:53 +02:00
parent c725fcb8df
commit 19390f8d4f
6 changed files with 157 additions and 31 deletions

View File

@@ -9,17 +9,18 @@
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/imgproc/imgproc.hpp>
#include "utils.hpp"
#include "LayerData.hpp"
namespace fmri {
using std::string;
using std::vector;
class Simulator {
public:
typedef float DType;
Simulator(const string &model_file, const string &weights_file, const string &means_file = "");
vector<DType> simulate(const string &input_file);
vector<LayerData> simulate(const string &input_file);
private:
caffe::Net<DType> net;