Skip to content

Commit d95fdc7

Browse files
committed
Elementwise layers parameters initialization.
They are initialized only in functors, not in inherited fields.
1 parent a28b3f3 commit d95fdc7

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

modules/dnn/src/layers/elementwise_layers.cpp

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,9 @@ Ptr<_Layer> _Layer::create() { \
1414

1515
Ptr<ReLULayer> ReLULayer::create(double negativeSlope)
1616
{
17-
return Ptr<ReLULayer>(new ElementWiseLayer<ReLUFunctor>(ReLUFunctor(negativeSlope)));
17+
Ptr<ReLULayer> layer(new ElementWiseLayer<ReLUFunctor>(ReLUFunctor(negativeSlope)));
18+
layer->negativeSlope = negativeSlope;
19+
return layer;
1820
}
1921

2022
Ptr<TanHLayer> TanHLayer::create()
@@ -40,7 +42,11 @@ Ptr<BNLLLayer> BNLLLayer::create()
4042
Ptr<PowerLayer> PowerLayer::create(double power /*= 1*/, double scale /*= 1*/, double shift /*= 0*/)
4143
{
4244
const PowerFunctor f(power, scale, shift);
43-
return Ptr<PowerLayer>(new ElementWiseLayer<PowerFunctor>(f));
45+
Ptr<PowerLayer> layer(new ElementWiseLayer<PowerFunctor>(f));
46+
layer->power = power;
47+
layer->scale = scale;
48+
layer->shift = shift;
49+
return layer;
4450
}
4551

4652
////////////////////////////////////////////////////////////////////////////

0 commit comments

Comments
 (0)