File tree Expand file tree Collapse file tree 2 files changed +8
-6
lines changed Expand file tree Collapse file tree 2 files changed +8
-6
lines changed Original file line number Diff line number Diff line change @@ -117,6 +117,9 @@ class BaseConvolutionLayerImpl : public ConvolutionLayer
117
117
{
118
118
kernel_size.assign (1 , kernel_size[0 ]);
119
119
strides.assign (1 , strides[0 ]);
120
+ dilations.assign (1 , dilations[0 ]);
121
+ pads_begin.assign (1 , pads_begin[0 ]);
122
+ pads_end.assign (1 , pads_end[0 ]);
120
123
}
121
124
CV_Assert (weightShape.dims () == kernel_size.size () + 2 );
122
125
for (int i = 0 ; i < kernel_size.size (); i++) {
Original file line number Diff line number Diff line change @@ -167,14 +167,13 @@ class PoolingLayerImpl CV_FINAL : public PoolingLayer
167
167
168
168
if (inputs[0 ].dims == 3 )
169
169
{
170
- // Pool1D
171
- kernel_size.erase (kernel_size. begin () + 1 );
172
- strides.erase (strides. begin () + 1 );
173
- pads_begin.erase (pads_begin. begin () + 1 );
174
- pads_end.erase (pads_end. begin () + 1 );
170
+ // Pool1D
171
+ kernel_size.assign ( 1 , kernel_size[ 0 ] );
172
+ strides.assign ( 1 , strides[ 0 ] );
173
+ pads_begin.assign ( 1 , pads_begin[ 0 ] );
174
+ pads_end.assign ( 1 , pads_end[ 0 ] );
175
175
}
176
176
177
-
178
177
#ifdef HAVE_OPENCL
179
178
poolOp.release ();
180
179
#endif
You can’t perform that action at this time.
0 commit comments