?? houghline.cpp
字號:
#include "dimage.h"
#include <math.h>
#define houghthresh 10
void diedaiyz(BYTE *image[2000],int w,int h);
void sanfenyz(BYTE *image[2000],int w,int h);
void sobel(BYTE *image_in[2000],int w,int h,BYTE *image_out[2000]);
void lapla1(BYTE *image_in[2000],int w,int h,BYTE *image_out[2000]);
void lapla2(BYTE *image_in[2000],int w,int h,BYTE *image_out[2000]);
void zzlb(BYTE *image_in[2000],int w,int h,BYTE *image_out[2000]);
void houghtrans1(BYTE *image_in[2000],int w,int h);
void houghtrans2(BYTE *image_in[2000],int w,int h);
void makeline1(BYTE *image_in[2000],int w,int h,double t,int k,int r,BYTE *image_out[2000]);
void makeline2(BYTE *image_in[2000],int w,int h,double t,int k,int r,BYTE *image_out[2000]);
int medianvalue(int *value);
int main()
{
int w=512,h=512;
Welcom();
BYTE *image[2000];
InitImage(image,w,h);
ReadRawFile(image,w,h,"D:\\peacock.raw");
//圖像讀取的路徑為：D:\\peacock.raw ，在保存的時候存在了E盤
//且均以peacock_后綴.raw命名，例如：peacock_lapla2.raw,peacock_hough.raw
//我們將所有的函數按不同的順序進行組合，以期得到最好的邊緣提取效果
// diedaiyz(image,w,h); //迭代法選取閾值，二值化，并將二值化后的圖像保存
// sanfenyz(image,w,h); //T=max-(max-min)/3，二值化，并將二值化后的圖像保存
// BYTE *image2[2000];
// InitImage(image2,w,h);
// zzlb(image,w,h,image2); //中值濾波，并將濾波后的圖像保存
//前面的為輸入數組，如image;后面的為輸出數組，如image2。以下均同，請注意之！
BYTE *image1[2000];
InitImage(image1,w,h);
// sobel(image,w,h,image1); //sobel算子對圖像進行梯度銳化，并將梯度銳化后的圖像保存
// lapla1(image,w,h,image1); //拉普拉斯算子，模板一：{0,-1,0,-1,4,-1,0,-1,0}，并將梯度銳化后的圖像保存
lapla2(image,w,h,image1); //拉普拉斯算子，模板二：{-1,-1,-1,-1,8,-1, -1,-1,-1}，并將梯度銳化后的圖像保存
// BYTE *image2[2000];
// InitImage(image2,w,h);
// zzlb(image1,w,h,image2); //中值濾波
// diedaiyz(image1,w,h); //迭代法選取閾值，二值化
sanfenyz(image1,w,h); //T=max-(max-min)/3，二值化
// BYTE *image2[2000];
// InitImage(image2,w,h);
// zzlb(image1,w,h,image2); //中值濾波
// houghtrans1(image1,w,h); //hough變換第一種算法：5x5模板
houghtrans2(image1,w,h); //hough變換第二種算法：循環hough,多次提取最長直線
return 1;
}
void houghtrans1(BYTE *image_in[2000],int w,int h) //hough變換第一種算法
{
double t=3.1415926;
t=t/180;
int rmax=w+h+1;
int *hough[180];
int k,r;
for(k=0;k<180;k++)
hough[k]=(int *)new int[rmax];
for(k=0;k<180;k++)
for(r=0;r<rmax;r++)
hough[k][r]=0;
int i,j;
for(i=0;i<h;i++) //將大于0的點標記為255，即白點
//自己繪制的圖像中可能存在一些點既不是0也不是255，另外為了避免由于前期處理不當導致hough失敗
for(j=0;j<w;j++)
{if(image_in[i][j]>0)image_in[i][j]=255;}
for(i=0;i<h;i++) //對原圖中的白點做hough變換
for(j=0;j<w;j++)
{
if(image_in[i][j]==255)
{
for(k=0;k<180;k++)
{
r=(int)(j*cos(k*t)+i*sin(k*t));
if(r<0)r+=w+h+1;
hough[k][r]++;
}
}
}
BYTE *image_out[2000];
InitImage(image_out,w,h);
for(k=0;k<180;k++) //尋找數組hough[][]的各個峰值，采用5x5模板算法
for(r=0;r<rmax;r++)
{
if(hough[k][r]>houghthresh)
{
int tag=1;
int dk,dr,dkt,drt;
for(dk=k-2;dk<=k+2;dk++)
{
for(dr=r-2;dr<=r+2;dr++)
{
if(dk<0)dkt=dk+180;
else if(dk>=180)dkt=dk-180;
else dkt=dk;
if(dr<0)drt=dr+w+h+1;
else drt=dr;
if(drt<rmax&&hough[dkt][drt]>hough[k][r])
{tag=0;break;}
}
if(tag==0)break;
}
if(tag==1)
{
// makeline1(image_in,w,h,t,k,r,image_out); //尋找到一條直線立即繪制
makeline2(image_in,w,h,t,k,r,image_out);
}
}
}
SaveRawFile(image_out,w,h,"E:\\peacock_hough.raw");
}
void houghtrans2(BYTE *image_in[2000],int w,int h) //hough變換第二種算法
{
BYTE *image_out[2000];
InitImage(image_out,w,h);
double t=3.1415926;
t=t/180;
int rmax=w+h+1;
int *hough[180];
int k,r;
for(k=0;k<180;k++)
hough[k]=(int *)new int[rmax];
int i,j;
for(i=0;i<h;i++)
for(j=0;j<w;j++)
if(image_in[i][j]>0)image_in[i][j]=255;//此處以上與算法一相同
while(1) //對hough變換的全部過程做循環
{
for(k=0;k<180;k++)
for(r=0;r<rmax;r++)
hough[k][r]=0;
for(i=0;i<h;i++) //對原圖作hough變換，同算法一
for(j=0;j<w;j++)
{
if(image_in[i][j]==255)
{
for(k=0;k<180;k++)
{
r=(int)(j*cos(k*t)+i*sin(k*t));
if(r<0)r+=w+h+1;
hough[k][r]++;
}
}
}
int max=0,max_k,max_r; //尋找最長的一條直線
for(k=0;k<180;k++)
for(r=0;r<rmax;r++)
{
if(hough[k][r]>max)
{
max=hough[k][r];
max_k=k;
max_r=r;
}
}
if(max<houghthresh)break; //當最大值小于設定的臨界值時，循環結束
// makeline1(image_in,w,h,t,max_k,max_r,image_out);
makeline2(image_in,w,h,t,max_k,max_r,image_out);
}
SaveRawFile(image_out,w,h,"E:\\peacock_hough.raw");
}
//繪制直線算法一
void makeline1(BYTE *image_in[2000],int w,int h,double t,int k,int r,BYTE *image_out[2000])
{
int i,j,temp;
for(i=0;i<h;i++) //重新對數組上的白點作hough變換，如果發現某點落在(k,r)的區域內，在原圖與新圖中均做標記。
for(j=0;j<w;j++)
{
if(image_in[i][j]==255)
{
temp=(int)(j*cos(k*t)+i*sin(k*t));
if(temp<0)temp+=w+h+1;
if(temp==r)
{image_in[i][j]=250;image_out[i][j]=255;}
}
}
}
//繪制圖像算法二
void makeline2(BYTE *image_in[2000],int w,int h,double t,int k,int r,BYTE *image_out[2000])
{
int i,j,temp;
for(i=0;i<h;i++) //對原圖重新作hough變換，標記已經提取到的直線(k,r)上的白點
for(j=0;j<w;j++)
{
if(image_in[i][j]==255)
{
temp=(int)(j*cos(k*t)+i*sin(k*t));
if(temp<0)temp+=w+h+1;
if(temp==r)
{image_in[i][j]=250;}
}
}
for(j=0;j<w;j++) //在輸出數組image_out[][]中通過逆函數繪制直線
{
if(k!=0)
{
if(k>90&&k<180&&r>h)r-=w+h+1;
i=(int)(r/sin(k*t)-j/tan(k*t));
if(i>=0&&i<h)image_out[i][j]=255;
}
}
}
//以下為二值化，梯度銳化，中值濾波等函數，不做注釋！
void diedaiyz(BYTE *image[2000],int w,int h)
{
int i,j;
int count[256];
for(i=0;i<256;i++)
count[i]=0;
for(i=0;i<h;i++)
for(j=0;j<w;j++)
count[image[i][j]]++;
int T=100;
while(1)
{
int temp=T;
int t1=0,t2=0,t3=0,t4=0;
for(i=0;i<=temp;i++)
{t1+=count[i];t3+=count[i]*i;}
for(i=temp+1;i<256;i++)
{t2+=count[i];t4+=count[i]*i;}
T=(int)((t3/t1+t4/t2)/2);
if(T==temp)break;
}
for(i=0;i<h;i++)
for(j=0;j<w;j++)
if(image[i][j]>=T)image[i][j]=255;
else image[i][j]=0;
SaveRawFile(image,w,h,"E:\\peacock_diedaiyz.raw");
}
void sanfenyz(BYTE *image[2000],int w,int h)
{
int i,j;
int fmax=0,fmin=255;
for(i=0;i<h;i++)
for(j=0;j<w;j++)
{
if(image[i][j]>fmax)fmax=image[i][j];
if(image[i][j]<fmin)fmin=image[i][j];
}
int T=fmax-(fmax-fmin)/3;
for(i=0;i<h;i++)
for(j=0;j<w;j++)
if(image[i][j]>=T)image[i][j]=255;
else image[i][j]=0;
SaveRawFile(image,w,h,"E:\\peacock_sanfenyz.raw");
}
void sobel(BYTE *image_in[2000],int w,int h,BYTE *image_out[2000])
{
int buf,buf1,buf2;
int i,j;
for(i=0;i<h;i++)
for(j=0;j<w;j++)
{
if(i-1>=0&&j-1>=0&&i+1<h&&j+1<w)
{
//x方向
buf1=(int)image_in[i-1][j+1]+2*(int)image_in[i][j+1]+(int)image_in[i+1][j+1];
buf1=buf1-(int)image_in[i-1][j-1]-2*(int)image_in[i][j-1]-(int)image_in[i+1][j-1];
//y方向
buf2=(int)image_in[i-1][j-1]+2*(int)image_in[i-1][j]+(int)image_in[i-1][j+1];
buf2=buf2-(int)image_in[i+1][j-1]-2*(int)image_in[i+1][j]-(int)image_in[i+1][j+1];
buf=abs(buf1)+abs(buf2);
if(buf>255)buf=255;
if(buf<0)buf=0;
image_out[i][j]=(BYTE)buf;
}
else
image_out[i][j]=(BYTE)0;
}
SaveRawFile(image_out,w,h,"E:\\peacock_sobel.raw");
}
void lapla1(BYTE *image_in[2000],int w,int h,BYTE *image_out[2000])
{
int buf;
int i,j;
for(i=0;i<h;i++)
for(j=0;j<w;j++)
{
if(i-1>=0&&j-1>=0&&i+1<h&&j+1<w)
{
//x方向
buf=4*(int)image_in[i][j]-(int)image_in[i-1][j]-(int)image_in[i][j-1]
-(int)image_in[i][j+1]-(int)image_in[i+1][j];
if(buf>255)buf=255;
if(buf<0)buf=0;
image_out[i][j]=(BYTE)buf;
}
else
image_out[i][j]=(BYTE)0;
}
SaveRawFile(image_out,w,h,"E:\\peacock_lapla1.raw");
}
void lapla2(BYTE *image_in[2000],int w,int h,BYTE *image_out[2000])
{
int buf;
int i,j;
for(i=0;i<h;i++)
for(j=0;j<w;j++)
{
if(i-1>=0&&j-1>=0&&i+1<h&&j+1<w)
{
//x方向
buf=8*(int)image_in[i][j]-(int)image_in[i-1][j-1]-(int)image_in[i-1][j]-(int)image_in[i-1][j+1]
-(int)image_in[i][j-1]-(int)image_in[i][j+1]-(int)image_in[i+1][j-1]
-(int)image_in[i+1][j]-(int)image_in[i+1][j+1];
if(buf>255)buf=255;
if(buf<0)buf=0;
image_out[i][j]=(BYTE)buf;
}
else
image_out[i][j]=(BYTE)0;
}
SaveRawFile(image_out,w,h,"E:\\peacock_lapla2.raw");
}
void zzlb(BYTE *image_in[2000],int w,int h,BYTE *image_out[2000])
{
int value[9];
int i,j;
for(i=1;i<h-1;i++)
for(j=1;j<w-1;j++)
{
value[0]=image_in[i-1][j-1];
value[1]=image_in[i-1][j];
value[2]=image_in[i-1][j+1];
value[3]=image_in[i][j-1];
value[4]=image_in[i][j];
value[5]=image_in[i][j+1];
value[6]=image_in[i+1][j-1];
value[7]=image_in[i+1][j];
value[8]=image_in[i+1][j+1];
image_out[i][j]=medianvalue(value);
}
SaveRawFile(image_out,w,h,"E:\\peacock_zzlb.raw");
}
int medianvalue(int *value)
{
int i,j,temp;
for(i=0;i<=7;i++)
for(j=0;j<=7-i;j++)
{
if(value[j]>value[j+1])
{
temp=value[j+1];
value[j+1]=value[j];
value[j]=temp;
}
}
return value[4];
}
?? 快捷鍵說明
復制代碼
Ctrl + C
搜索代碼
Ctrl + F
全屏模式
F11
切換主題
Ctrl + Shift + D
顯示快捷鍵
?
增大字號
Ctrl + =
減小字號
Ctrl + -