Anatoly Baksheev
10 years ago
parent
42b36fb346
commit
10c3a1378a
4 changed files with 185 additions and 172 deletions
@ -0,0 +1,177 @@ |
||||
#ifndef __OPENCV_DNN_DNN_HPP__ |
||||
#define __OPENCV_DNN_DNN_HPP__ |
||||
|
||||
#include <opencv2/core.hpp> |
||||
#include <map> |
||||
#include <vector> |
||||
#include <opencv2/dnn/dict.hpp> |
||||
|
||||
namespace cv |
||||
{ |
||||
namespace dnn |
||||
{ |
||||
class Layer; |
||||
class NetConfiguration; |
||||
class Net; |
||||
class Blob; |
||||
class LayerParams; |
||||
|
||||
//wrapper over cv::Mat and cv::UMat
|
||||
class CV_EXPORTS Blob |
||||
{ |
||||
public: |
||||
Blob(); |
||||
Blob(InputArray in); |
||||
|
||||
void fill(InputArray in); |
||||
void fill(int ndims, const int *sizes, int type, void *data, bool deepCopy = true); |
||||
void create(int ndims, const int *sizes, int type = CV_32F); |
||||
|
||||
bool empty() const; |
||||
|
||||
Mat& getMatRef(); |
||||
const Mat& getMatRef() const; |
||||
Mat getMat(); |
||||
Mat getMat(int num, int channel); |
||||
|
||||
int cols() const; |
||||
int rows() const; |
||||
Size size() const; |
||||
|
||||
int channels() const; |
||||
int num() const; |
||||
|
||||
Vec4i shape() const; |
||||
|
||||
private: |
||||
Mat m; |
||||
}; |
||||
|
||||
class CV_EXPORTS LayerParams : public Dict |
||||
{ |
||||
public: |
||||
|
||||
std::vector<Blob> learnedBlobs; |
||||
}; |
||||
|
||||
class CV_EXPORTS LayerRegister |
||||
{ |
||||
public: |
||||
|
||||
typedef Layer* (*Constuctor)(); |
||||
|
||||
static void registerLayer(const String &type, Constuctor constructor); |
||||
|
||||
static void unregisterLayer(const String &type); |
||||
|
||||
static Ptr<Layer> createLayerInstance(const String &type); |
||||
|
||||
private: |
||||
LayerRegister(); |
||||
LayerRegister(const LayerRegister &lr); |
||||
|
||||
static std::map<String, Constuctor> registeredLayers; |
||||
}; |
||||
|
||||
//this class allows to build new Layers
|
||||
class CV_EXPORTS Layer |
||||
{ |
||||
public: |
||||
|
||||
//TODO: this field must be declared as public if we want support possibility to change these params in runtime
|
||||
std::vector<Blob> learnedParams; |
||||
|
||||
virtual ~Layer(); |
||||
|
||||
//type of Layer
|
||||
virtual String type() const; |
||||
|
||||
//setUp calls once (think that it's constructor)
|
||||
virtual void setUp(LayerParams ¶ms); |
||||
|
||||
//after setUp the following two function must be able to return values
|
||||
virtual int getNumInputs(); |
||||
virtual int getNumOutputs(); |
||||
|
||||
//maybe useless function
|
||||
//shape of output blobs must be adjusted with respect to shape of input blobs
|
||||
virtual void adjustShape(const std::vector<Blob> &inputs, std::vector<Blob> &outputs); |
||||
|
||||
virtual void forward(std::vector<Blob> &inputs, std::vector<Blob> &outputs); |
||||
}; |
||||
|
||||
//TODO: divide NetConfiguration interface and implementation, hide internal data
|
||||
//TODO: maybe eliminate all int ids and replace them by string names
|
||||
//Proxy class for different formats
|
||||
//Each format importer must populate it
|
||||
class CV_EXPORTS NetConfiguration |
||||
{ |
||||
public: |
||||
|
||||
static Ptr<NetConfiguration> create(); |
||||
|
||||
int addLayer(const String &name, const String &type); |
||||
|
||||
void deleteLayer(int layerId); |
||||
|
||||
void setLayerParams(int layerId, LayerParams ¶ms); |
||||
|
||||
//each output of each layer can be labeled by unique string label (as in Caffe)
|
||||
//if label not specified then %layer_name%:c_%N% will be used
|
||||
void setLayerOutputLabels(int layerId, const std::vector<String> &outputNames); |
||||
|
||||
//version #1
|
||||
void addConnection(int fromLayer, int fromLayerOutput, int toLayer, int toLayerInput); |
||||
|
||||
//or maybe version #2
|
||||
inline int getBlobId(int layerId, int inputOutputNumber) |
||||
{ |
||||
return (layerId << 16) + inputOutputNumber; |
||||
} |
||||
|
||||
void addConnection(int outputId, int inputId); |
||||
|
||||
void addConnections(const std::vector<int> &outputIds, const std::vector<int> &inputIds); |
||||
|
||||
private: |
||||
|
||||
int lastLayerId; |
||||
std::map< int, Ptr<Layer> > layers; |
||||
std::map< int, std::vector<String> > layerOutputLabels; |
||||
}; |
||||
|
||||
|
||||
class CV_EXPORTS Net |
||||
{ |
||||
public: |
||||
|
||||
static Ptr<Net> create(Ptr<NetConfiguration> config); |
||||
|
||||
virtual ~Net() = 0; |
||||
|
||||
virtual int getBlobId(int layerId, int outputId) = 0; |
||||
|
||||
virtual int getBlobId(const String &blobName) = 0; |
||||
|
||||
virtual void forward(std::vector< int, Ptr<Blob> > &inputBlobs, std::vector<int, Ptr<Blob> > &outputBlobs) = 0; |
||||
|
||||
virtual void forward(int layer, std::vector<Ptr<Blob> > &layerOutputs) = 0; |
||||
}; |
||||
|
||||
class CV_EXPORTS Importer |
||||
{ |
||||
public: |
||||
|
||||
virtual void populateNetConfiguration(Ptr<NetConfiguration> config) = 0; |
||||
|
||||
virtual ~Importer(); |
||||
}; |
||||
|
||||
CV_EXPORTS Ptr<Importer> createCaffeImporter(const String &prototxt, const String &caffeModel); |
||||
|
||||
} |
||||
} |
||||
|
||||
#include <opencv2/dnn/dnn.inl.hpp> |
||||
|
||||
#endif /* __OPENCV_DNN_DNN_HPP__ */ |
||||
Loading…
Reference in new issue