图像配准拼接算法_图像配准算法c++代码-CSDN博客


struct ImageInfo
{
	//图片内容和坐标等数据
};
struct Shift
{
	int x, y;//两张图片的配准结果
};
struct Pos
{
	int x, y;//小图坐标
};

class RegistrationStitch
{
public:
	RegistrationStitch(int row, int col, vector<ImageInfo>images) :row{ row }, col{ col }, images{ images }{}
	ImageInfo GetStitchImage()
	{
		Registration();
		vector<Pos> allPos = GetPos(shiftLR, shiftUD);
		return GetStitchImage(images, allPos);
	}
private:
	int GetId(int r, int c) //所有小图编号0到row*col-1
	{
		return r * col + c;
	}
	void Registration()//配准算法
	{
		for (int r = 0; r < row; r++) {
			for (int c = 0; c < col; c++) {
				if (r)shiftUD[GetId(r, c)] = Registration(images[GetId(r - 1, c)], images[GetId(r, c)]);
				if (c)shiftLR[GetId(r, c)] = Registration(images[GetId(r, c - 1)], images[GetId(r, c)]);
			}
		}
	}
	Shift Registration(ImageInfo image1, ImageInfo image2)//两张小图配准
	{
		return Shift{};
	}
	static vector<Pos> GetPos(map<int, Shift>& shiftLR, map<int, Shift>& shiftUD)
	{
		return vector<Pos>{};
	}
	static ImageInfo GetStitchImage(vector<ImageInfo>& images, vector<Pos>& allPos)
	{
		ImageInfo ans;
		for (int i = 0; i < images.size(); i++) {
			// ans = CopyImage(ans,images[i],allPos[i]);
		}
		return ans;
	}
private:
	int row, col;
	vector<ImageInfo>images;
	map<int, Shift> shiftLR;
	map<int, Shift> shiftUD;
}

这里的GetPos函数是static的，即只需要入参不需要图片等信息。其他私有静态成员函数同理。

GetPos里面省略的是一个BFS或者DFS算法，也可以用带权并查集。

2，减少配准

上面的伪代码进行了所有配准，其实只需要一部分。

以3*3为例，不需要做12次配准，只需要做8次配准。

不是任意8个配准就行，其实就是从一个9个点12条边的图中选出任意一颗生成树。

既然配准算法是绝对准确可靠的，那么任选一颗生成树，结果都是一样的。

其实可以选择一个固定方式的生成树，如上图（所有的左右配准加第一列的上下配准）

3，实时配准

如果想加快拼接出图的速度，可以一边扫描小图一边完成配准。


struct ImageInfo
{
	//图片内容和坐标等数据
};
struct Shift
{
	int x, y;//两张图片的配准结果
};
struct Pos
{
	int x, y;//小图坐标
};

class RegistrationStitch
{
public:
	RegistrationStitch(int row, int col) :row{ row }, col{ col }{}
	void Push(ImageInfo image)
	{
		images.push_back(image);
		Registration(); //每次传入一张小图，调用一次配准
	}
	ImageInfo GetStitchImage()
	{
		vector<Pos> allPos = GetPos(shiftLR, shiftUD);
		return GetStitchImage(images, allPos);
	}
private:
	int GetId(int r, int c) //所有小图编号0到row*col-1
	{
		return r * col + c;
	}
	void Registration()//配准算法
	{
		vector<pair<int, int>> pairs;
		// pairs = ... 计算当前传入小图和哪些小图需要配准
		for (auto p : pairs) {
			auto shift = Registration(images[p.first], images[p.second]);
			// shiftLR or shiftUD 存入结果shift
		}
	}
	Shift Registration(ImageInfo image1, ImageInfo image2)//两张小图配准
	{
		return Shift{};
	}
	static vector<Pos> GetPos(map<int, Shift>& shiftLR, map<int, Shift>& shiftUD)
	{
		return vector<Pos>{};
	}
	static ImageInfo GetStitchImage(vector<ImageInfo>& images, vector<Pos>& allPos)
	{
		ImageInfo ans;
		for (int i = 0; i < images.size(); i++) {
			// ans = CopyImage(ans,images[i],allPos[i]);
		}
		return ans;
	}
private:
	int row, col;
	vector<ImageInfo>images;
	map<int, Shift> shiftLR;
	map<int, Shift> shiftUD;
}