首页 > 编程语言 > 详细

C++基于armadillo col2im的实现

时间:2021-05-23 23:18:18      阅读:28      评论:0      收藏:0      [点我收藏+]

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;
}

C++基于armadillo col2im的实现

原文:https://www.cnblogs.com/wyblikeswsf/p/14801505.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!