图像模式识别之几何分类器(一)

于 2013-07-07 10:36:29 发布
阅读量240 收藏
点赞数
分类专栏： 机器学习 文章标签： 模式识别
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/diarymaggie/article/details/9262291
版权 
#include "cv.h"
#include "highgui.h"
#include "utils.h"

#include<stack>

using namespace cv;
using namespace std;

#define PI 3.1415926
#define RADIAN(angle) ((angle)*PI/180.0)

struct pattern
{
	int number; //该类型数字样品个数
	double feature[200][25];//该类型数字样品特征，每个特征为25维
	pattern()
	{
		number=0;
	}
};

struct Position
{
	int bottom;
	int top;
	int left;
	int right;
};

pattern ptns[10]; //一共10个数字
double testsample[25]; //手写数字的特征 25维

//判断手写数字可否保存为cls类别的一个样品 同一类别不能有重复
bool savable(int cls)
{
	for(int i=0;i<ptns[cls].number;i++)
	{
		bool flag=true;
		for(int j=0;j<25;j++)
			flag&=(fabs(ptns[cls].feature[i][j]-testsample[j])<0.0000001 ? true:false);
		if(flag) return false;
	}

	return true;
}

//将手写数字保存为cls类别中的一个样品
void Save(int cls)
{
	for(int i=ptns[cls].number;i>0;i--)
		for(int j=0;j<25;j++)
			ptns[cls].feature[i][j]=ptns[cls].feature[i-1][j];

	for(int j=0;j<25;j++)
		ptns[cls].feature[0][j]=testsample[j];

	ptns[cls].number++;
}

//计算手写数字的在图片中出现的位置
Position GetPosition(Mat& mat)
{
	int M=mat.rows;
	int N=mat.cols;

	bool flag=false;
	int i,j;
	Position ret;

	//top
	for(i=0;i<M-1;i++)
	{
		flag=false;
		for(j=0;j<N;j++)
		{
			if(mat.at<uchar>(i,j)<128)
			{
				flag=true;
				break;
			}
		}
		if(flag) break;
	}
	ret.top=i;

	//bottom
	for(i=M-1;i>0;i--)
	{
		flag=false;
		for(j=0;j<N;j++)
		{
			if(mat.at<uchar>(i,j)<128)
			{
				flag=true;
				break;
			}
		}
		if(flag) break;
	}
	ret.bottom=i;

	//left
	for(j=0;j<N-1;j++)
	{
		flag=false;
		for(i=0;i<M;i++)
		{
			if(mat.at<uchar>(i,j)<128)
			{
				flag=true;
				break;
			}
		}
		if(flag) break;
	}
	ret.left=j;

	for(j=N-1;j>0;j--)
	{
		flag=false;
		for(i=0;i<M;i++)
		{
			if(mat.at<uchar>(i,j)<128)
			{
				flag=true;
				break;
			}
		}
		if(flag) break;
	}
	ret.right=j;

	return ret;
}

double Cal(int row,int col,Position pos,Mat& mat)
{
	int w,h;
	w=(pos.right-pos.left)/5;
	h=(pos.bottom-pos.top)/5;

	int count=0;

	for(int i=pos.top+row*h;i<pos.top+(row+1)*h;i++)
	{
		for(int j=pos.left+col*w;j<pos.left+(col+1)*w;j++)
		{
			if(mat.at<uchar>(i,j)<128)
				count++;
		}
	}
	return ((double)count)/(w*h);
}
//将手写数字特征保存在变量testsample中
void SetFeature(Mat& mat,Position pos)
{
	for(int i=0;i<5;i++)
	{
		for(int j=0;j<5;j++)
		{
			testsample[5*i+j]=Cal(i,j,pos,mat);
		}
	}
}

//奖惩算法
int JiangCheng()
{
	double w[10][26];
	double d[10];
	double x[26];
	double hx[10];
	bool flag;
	int c=0;

	for(int n=0;n<10;n++)
		for(int i=0;i<26;i++)
			w[n][i]=0;

	do
	{
		flag=true;
		for(int n=0;n<10;n++)
		{
			for(int i=0;i<ptns[n].number;i++)
			{
				for(int j=0;j<25;j++)
					x[j]=ptns[n].feature[i][j];
				x[25]=1;

				for(int j=0;j<10;j++)
					d[j]=0;
				for(int j=0;j<10;j++)
					for(int k=0;k<26;k++)
						d[j]+=w[j][k]*x[k];
				bool f=true;
				//判断d[n]是否是最大值
				for(int j=0;j<10;j++)
					if(j!=n)
						f&=(d[n]>d[j]?true:false);

				if(f) flag&=true;
				else //修正权值
				{
					for(int j=0;j<10;j++)
					{
						for(int k=0;k<26;k++)
						{
							if(j==n)
								w[j][k]+=x[k];
							else if(d[j]>d[n])
								w[j][k]-=x[k];
						}
					}
					flag&=false;
				}
			}
		}
		if(++c>2000) break;
	}while(!flag);

	double num[26];
	for(int i=0;i<25;i++)
		num[i]=testsample[i];
	num[25]=1;

	//判别函数
	for(int n=0;n<10;n++)
	{
		hx[n]=0;
		for(int i=0;i<26;i++)
			hx[n]+=w[n][i]*num[i];
	}

	double maxVal=hx[0];
	int ret=0;
	for(int n=0;n<10;n++)
	{
		if(hx[n]>maxVal)
		{
			maxVal=hx[n];
			ret=n;
		}
	}

	return ret;
}

//增量校正算法
int ZengLiangJiaoZheng()
{
	double w[10][26]; //权值
	double d[10]; //迭代计算结果权矢量
	double x[26]; //样本
	
	//权值初值为0
	for(int n=0;n<10;n++)
		for(int i=0;i<26;i++)
			w[n][i]=0.0;

	bool flag;
	int c=0,cc=1;
	do
	{
		flag=true;
		for(int n=0;n<10;n++)
		{
			for(int i=0;i<ptns[n].number;i++)
			{
				//取每一个样本
				for(int j=0;j<25;j++)
					x[j]=ptns[n].feature[i][j];
				x[25]=1;

				//计算结果权矢量
				for(int j=0;j<10;j++)
					d[j]=0;

				for(int j=0;j<10;j++)
					for(int k=0;k<26;k++)
						d[j]+=w[j][k]*x[k];

				//判断d[n]是否是最大值
				bool f=true;
				for(int j=0;j<10;j++)
					if(j!=n)
						f&=(d[n]>d[j])?true:false;
				if(f) flag&=true;
				else flag&=false;

				//增量修改权值
				for(int j=0;j<10;j++)
				{
					double rX;
					if(j==n) rX=1.0;
					else rX=0.0;

					if(d[j]<rX) //W(k)X(k)<r[X(k)]
					{
						for(int k=0;k<26;k++)
							w[j][k]+=x[k]/cc;
					}
					else //W(k)X(k)>=r[X(k)]
					{
						for(int k=0;k<26;k++)
							w[j][k]-=x[k]/cc;
					}
				}
				cc++;
			}
		}
		if(++c>2000) break;
	}while(!flag);

	double hx[10];
	double num[26];
	for(int i=0;i<25;i++)
		num[i]=testsample[i];
	num[25]=1;

	for(int n=0;n<10;n++)
	{
		hx[n]=0;
		for(int i=0;i<26;i++)
			hx[n]+=w[n][i]*num[i];
	}

	double maxVal=hx[0];
	int ret=0;

	for(int n=1;n<10;n++)
	{
		if(hx[n]>maxVal)
		{
			maxVal=hx[0];
			ret=n;
		}
	}
	return ret;
}

//LMSE方法验证可分性
bool IsClassifiedByLMSE(int class0,int class1)
{
	double X[40][26];
	double XSharp[26][40];
	double w[26];
	double XW[40]; //X*W
	double b[40];
	double e[40];

	bool flag,flag2;
	int c=1;

	//每类各取20个样品 构成样品矩阵X
	for(int i=0;i<20;i++)
	{
		for(int j=0;j<25;j++)
			X[i][j]=ptns[class0].feature[i][j];
		X[i][25]=1.0;
	}
	for(int i=20;i<40;i++)
	{
		for(int j=0;j<25;j++)
			X[i][j]=ptns[class1].feature[i][j];
		X[i][25]=1.0;
	}

	//求X的规范逆矩阵
	NormalizeMatrixInv(X,XSharp);

	//b
	for(int i=0;i<40;i++)
		b[i]=1;

	do
	{
		double maxVal=0,minVal=INT_MAX;
		//w=x#*b
		brmul(XSharp,b,w);
		brmul(X,w,XW);

		flag=flag2=true;

		for(int i=0;i<40;i++)
		{
			e[i]=XW[i]-b[i];
			flag&=e[i]<0?true:false;//e的分量是否全负
		}

		for(int i=0;i<40;i++)
		{
			double tempE=fabs(e[i]);
			if(maxVal<tempE)
				maxVal=tempE;
			if(minVal>tempE)
				minVal=tempE;
		}
		if(minVal>10000000)
			flag2=false;
		else if(maxVal<0.000001)
			flag2=false;

		//调整b
		if(!flag)
			for(int i=0;i<40;i++)
				b[i]+=(e[i]+fabs(e[i]));
		
		if(!flag2) return true;;

		if(++c>300) break;
	}while(!flag);

	return false;
}

int LMSE()
{
	double w[10][26];
	double d[10];
	double x[26];
	bool flag;

	for(int n=0;n<10;n++)
		for(int i=0;i<26;i++)
			w[n][i]=0;
	int c=0,cc=1;
	do
	{
		flag=true;
		for(int n=0;n<10;n++)
		{
			for(int i=0;i<ptns[n].number;i++)
			{
				for(int j=0;j<25;j++)
					x[j]=ptns[n].feature[i][j];
				x[25]=1.0;

				for(int j=0;j<10;j++)
					d[j]=0;
				for(int j=0;j<10;j++)
					for(int k=0;k<26;k++)
						d[j]+=w[j][k]*x[k];
				bool f=true;
				for(int j=0;j<10;j++)
					if(j!=n)
						f&=(d[n]>d[j]?true:false);
				if(f) flag&=true;
				else flag&=false;

				//更新wi
				for(int j=0;j<10;j++)
				{
					double rx;
					if(j==n) rx=1;
					else rx=0;

					for(int k=0;k<26;k++)
						w[j][k]+=x[k]*(rx-d[j])/cc;
				}
			}
			++cc;
		}
		if(++c>5000) break;
	}while(!flag);

	double hx[10];
	double num[26];

	for(int i=0;i<25;i++)
		num[i]=testsample[i];
	num[25]=1.0;

	//判别函数
	for(int n=0;n<10;n++)
	{
		hx[n]=0;
		for(int i=0;i<26;i++)
			hx[n]+=w[n][i]*num[i];
	}

	double maxVal=hx[0];
	int ret=0;

	for(int n=1;n<10;n++)
		if(maxVal<hx[n])
		{
			maxVal=hx[n];
			ret=n;
		}

	return ret;
}

int main()
{
	//读取图像
	Mat image;

	//样本库特征提取
	int cls;
	char filename[100]="a.txt";
	char filePath[100];
	FILE* fp=fopen(filename,"r");
	while(fscanf(fp,"%d%s",&cls,filePath))
	{
		if(cls==-1)
			break;
		Mat image=imread(filePath,CV_LOAD_IMAGE_GRAYSCALE);
		Position pos=GetPosition(image);
		SetFeature(image,pos);
		if(savable(cls))
			Save(cls);
	}
	fclose(fp);

	image=imread("3.jpg",CV_LOAD_IMAGE_GRAYSCALE);
	//计算手写数字的位置
	Position pos=GetPosition(image);
	//计算手写数字的特征
	SetFeature(image,pos);

	printf("%d\n",LMSE());
	//cvNamedWindow("test",CV_WINDOW_AUTOSIZE);
	//imshow("test",image);

	waitKey(0);
	return 0;
}

diarymaggie
关注
专栏目录
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值