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

support ReduceLayer without reshape layer.

Browse Source
This commit is contained in:
Zihao Mu
2022-08-02 10:32:31 +08:00
parent 98c33c605d
commit d4640f4647
5 changed files with 55 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/dnn/include/opencv2/dnn/all_layers.hpp
+2 -1
View File
@@ -334,7 +334,8 @@ CV__DNN_INLINE_NS_BEGIN
{
public:
int reduceType;
std::vector<size_t> reduceDims;
// reduceDims contains the dimensions that need to be reduced, targetDims is the target output dimension.
std::vector<size_t> reduceDims, targetDims;
static Ptr<ReduceLayer> create(const LayerParams& params);
};