二维数组转Mat

二维数组转Mat

image 转Mat

cv::Mat image_to_mat(image img)
{
	int channels = img.c;
	int width = img.w;
	int height = img.h;
	cv::Mat mat = cv::Mat(height, width, CV_8UC(channels));
	int step = mat.step;

	for (int j = 0; j < h; ++j) {
		for (int i = 0; i< w; ++i) {
			for (int k = 0; k < c; ++k) {
				float val = img.data[k*h*w + j*w + i];
				mat.data[j*step + i*c + k] = (unsigned char)(val * 255);
			}
		}
	}
	return mat;
}

mat转image:

image mat_to_image(cv::Mat mat)
{
    int w = mat.cols;
    int h = mat.rows;
    int c = mat.channels();
    image im = make_image(w, h, c);
    unsigned char *data = (unsigned char *)mat.data;
    int step = mat.step;
    for (int y = 0; y < h; ++y) {
        for (int k = 0; k < c; ++k) {
            for (int x = 0; x < w; ++x) {
                //uint8_t val = mat.ptr<uint8_t>(y)[c * x + k];
                //uint8_t val = mat.at<Vec3b>(y, x).val[k];
                //im.data[k*w*h + y*w + x] = val / 255.0f;

                im.data[k*w*h + y*w + x] = data[y*step + x*c + k] / 255.0f;
            }
        }
    }
    return im;
}

一维数组转换为Mat

image convert_to_image(unsigned char* data,int w,int h,int c)
{
	
	int i, j, k;
	image im = make_image(w, h, c);
	for (k = 0; k < c; ++k) {
		for (j = 0; j < h; ++j) {
			for (i = 0; i < w; ++i) {
				int dst_index = i + w*j + w*h*k;
				int src_index = k + c*i + c*w*j;
				im.data[dst_index] = (float)data[src_index] / 255.;
				//printf("c:%d,h:%d,w:%d,src_index:%d,dst_index:%d\n", k, j, i, src_index, dst_index);
			}
		}
	}
	//free(data);
	return im;
}

单通道转换为三通道：

void channelsOneToThree(unsigned char* dst, unsigned char* src, int w, int h, int c)
{
	for (int i = 0; i < h; ++i)
	{
		for (int j = 0; j < w; ++j)
		{
			int iPos = j * c;
			int k = i * w * c + iPos;
			if (c == 3)
			{
				dst[k] = src[j * h + h - i - 1];
				dst[k + 1] = src[j * h + h - i - 1];
				dst[k + 2] = src[j * h + h - i - 1];

			}
		}
	}
}