Add a model loader/simulator that actually loads caffe models.

This commit is contained in:
2017-10-07 17:33:11 +02:00
parent 4622c7c8c4
commit a034d2a7f3
5 changed files with 54 additions and 5 deletions

18
src/Simulator.cpp Normal file
View File

@@ -0,0 +1,18 @@
#include <iostream>
#include <string>
#include "Simulator.hpp"
using namespace caffe;
using namespace std;
using namespace fmri;
Simulator::Simulator(const string& model_file, const string& weights_file) :
net(model_file, TEST)
{
net.CopyTrainedLayersFrom(weights_file);
}
void Simulator::simulate(const string& image_file)
{
cerr << "This is not implemented yet." << endl;
}