potok/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2026-07-29 15:23:05 +04:00
Code Packages Releases Wiki Activity

PReLU layer for multidimensional input

Browse Source
This commit is contained in:
Dmitry Kurtaev
2017-10-23 14:30:40 +03:00
parent cca99bf824
commit a36ebaecdc
4 changed files with 10 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/dnn/src/layers/elementwise_layers.cpp
+7 -8
View File
@@ -80,20 +80,19 @@ public:
void operator()(const Range &r) const
{
int nstripes = nstripes_, nsamples, outCn;
size_t planeSize;
int nstripes = nstripes_, nsamples = 1, outCn = 1;
size_t planeSize = 1;
if( src_->dims == 4 )
if (src_->dims > 1)
{
nsamples = src_->size[0];
outCn = src_->size[1];
planeSize = (size_t)src_->size[2]*src_->size[3];
}
else
{
nsamples = outCn = 1;
planeSize = (size_t)src_->total();
}
outCn = src_->size[0];
for (int i = 2; i < src_->dims; ++i)
planeSize *= src_->size[i];
size_t stripeSize = (planeSize + nstripes - 1)/nstripes;
size_t stripeStart = r.start*stripeSize;