C++基于armadillo col2im的实现

C++基于armadillo im2col的实现
现在是要使用col2im函数,同样的因为原理差不多,我就把im2col语句顺序改了改,就成功了。
原理肾么的就看上一篇吧。
我多增加了一个iff参数判断是不是滤波器,然后在函数中进行了转置,这样就可以安照通常的思维,横向展开。im2col函数代码也可以依此更改,我就不把im2col放上来了。

1. 代码

field<mat> col2im(mat col, mat input_shape ,int filter_h, int filter_w, int stride, int pad, bool iff)
{
	int N, C, H, W;
	N = input_shape(0);
	C = input_shape(1);
	H = input_shape(2);
	W = input_shape(3);
	int out_h = (H + 2 * pad - filter_h) / stride + 1;
	int out_w = (W + 2 * pad - filter_w) / stride + 1;
	field<mat> img(N, C);
	img.for_each([H, W](mat& X) {X = mat(H, W, fill::zeros); });
	for (int n = 0, z = 0; n < N; n++)
	{
		for (int i = 0; i < out_h; i++)
		{
			for (int j = 0; j < out_w; j++, z++)
			{
				for (int k = 0; k < C; k++)
				{
					mat filter(1, filter_h * filter_w, fill::zeros);
					int x = z;
					int y0 = filter_h * filter_w * k;
					int y1 = filter_h * filter_w * k + filter_h * filter_w - 1;
					filter = col(span(x, x), span(y0, y1));
					filter.reshape(filter_h, filter_w);
					img(n, k)(span(i * stride, i * stride + filter_h - 1), span(j * stride, j * stride + filter_w - 1)) = filter;
				}
			}
		}
	}
	if(!iff)
		img.for_each([](mat& X) {X = X.t(); });
	return img;
}
我在此签名
posted @ 2021-05-23 16:56  c艹用户  阅读(17)  评论(0编辑  收藏  举报