File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -215,12 +215,12 @@ class im2row_CpuPBody : public cv::ParallelLoopBody
215
215
int row_offset = in_r*width;
216
216
int out_col_offset = k_r*kw;
217
217
int in_c = start_in_c + start_k_c*dw;
218
-
218
+
219
219
for (int k_c = start_k_c; k_c < end_k_c; k_c++, in_c += dw)
220
220
{
221
221
int in_index = channels_offset + row_offset + in_c;
222
222
int out_index = out_row_offset + out_ch_offset + out_col_offset + k_c;
223
-
223
+
224
224
data_col_[out_index] = data_im_[in_index];
225
225
}
226
226
}
@@ -346,7 +346,7 @@ void col2im_cpu(const Dtype* data_col,
346
346
{
347
347
int h_pad = h * stride_h - pad_h + h_offset * dilation_h;
348
348
int w_pad = w * stride_w - pad_w + w_offset * dilation_w;
349
-
349
+
350
350
if (h_pad >= 0 && h_pad < height && w_pad >= 0 && w_pad < width)
351
351
data_im[(c_im * height + h_pad) * width + w_pad] +=
352
352
data_col[(c * height_col + h) * width_col + w];
You can’t perform that action at this time.
0 commit comments