potok/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2026-07-30 07:43:03 +04:00
Code Packages Releases Wiki Activity

Merge pull request #21601 from zihaomu:add_reduceLayer

Browse Source
This commit is contained in:
Alexander Alekhin
2022-03-26 10:20:10 +00:00
parent e5bdab0355 b6b5c27cec
commit c9b90884da
8 changed files with 796 additions and 197 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/dnn/include/opencv2/dnn/all_layers.hpp
+14
View File
@@ -325,6 +325,20 @@ CV__DNN_INLINE_NS_BEGIN
static Ptr<PoolingLayerInt8> create(const LayerParams& params);
};
class CV_EXPORTS ReduceLayer : public Layer
{
public:
int reduceType;
std::vector<size_t> reduceDims;
static Ptr<ReduceLayer> create(const LayerParams& params);
};
class CV_EXPORTS ReduceLayerInt8 : public ReduceLayer
{
public:
static Ptr<ReduceLayerInt8> create(const LayerParams& params);
};
class CV_EXPORTS SoftmaxLayer : public Layer
{
public: