Actually run the network.

This commit is contained in:
2017-10-07 21:35:59 +02:00
parent dbcb4ead51
commit cf08dd4761
3 changed files with 120 additions and 5 deletions

View File

@@ -2,11 +2,16 @@
#include <string>
#include <memory>
#include <vector>
#include <caffe/caffe.hpp>
#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/imgproc/imgproc.hpp>
namespace fmri {
using std::string;
using std::vector;
class Simulator {
public:
@@ -18,5 +23,10 @@ namespace fmri {
private:
caffe::Net<DType> net;
cv::Size input_geometry;
unsigned int num_channels;
vector<cv::Mat> getWrappedInputLayer();
cv::Mat preprocess(cv::Mat original) const;
};
}