Actually run the network.
This commit is contained in:
@@ -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;
|
||||
};
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user