Bert Peters
470399aabf
Contrary to layerdata, this describes the layer itself rather than the layer contents.
26 lines
536 B
C++
26 lines
536 B
C++
#pragma once
|
|
|
|
#include <string>
|
|
#include <memory>
|
|
#include <vector>
|
|
#include "LayerData.hpp"
|
|
#include "LayerInfo.hpp"
|
|
|
|
namespace fmri {
|
|
using std::string;
|
|
using std::vector;
|
|
|
|
class Simulator {
|
|
public:
|
|
Simulator(const string &model_file, const string &weights_file, const string &means_file = "");
|
|
~Simulator();
|
|
|
|
vector<LayerData> simulate(const string &input_file);
|
|
const std::map<std::string, LayerInfo>& layerInfo() const;
|
|
|
|
private:
|
|
struct Impl;
|
|
std::unique_ptr<Impl> pImpl;
|
|
};
|
|
}
|