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:
@@ -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;
|
||||
|
||||
Reference in New Issue
Block a user