機(jī)器視覺處理的測試程序
在機(jī)器視覺處理中,我們經(jīng)常要對(duì)檢測到的物體的方位特征進(jìn)行評(píng)估。比如說,我們要 OCR 識(shí)別一個(gè)字符串。那么這個(gè)字符串與x軸的夾角就很重要,我們需要這個(gè)信息把這個(gè)字符串轉(zhuǎn)正,然后才方便識(shí)別。
條形碼識(shí)別也類似,尤其是當(dāng)我們條形碼不是很清晰時(shí),首先將條形碼轉(zhuǎn)正,然后用各向異性的濾波器處理一下,可以讓條形碼變得更清晰易于讀取。
這里給出一種基于統(tǒng)計(jì)參數(shù)的特征提取方法。這個(gè)方法已經(jīng)有幾十年歷史了,算是個(gè)老方法,但是效果很不錯(cuò),所以值得寫篇文章來介紹介紹。
區(qū)域的矩
一片區(qū)域 R 的矩定義為:
當(dāng)p 和q 都取 0 時(shí),得到的就是這片區(qū)域的面積。也就是:
矩還可以歸一化,也就是用上面的定義再除以面積 a。
表示的是這片區(qū)域的重心。可以用它來描述區(qū)域的位置。
歸一化的矩回隨區(qū)域在圖像中的位置不同而變化,要去除這個(gè)影響,可以用中心矩,中心矩只反映區(qū)域本身的特征。
具體的方法是將這個(gè)區(qū)域當(dāng)作一個(gè)橢圓區(qū)域,那么用上面5個(gè)參量就可以計(jì)算出橢圓的長短軸和旋轉(zhuǎn)角度。具體公式如下:
橢圓的這幾個(gè)參數(shù)的圖形解釋如下圖:
利用這幾個(gè)參數(shù)就可以確定區(qū)域的方位和尺寸了。
比如我們有下面的一幅測試圖像。
用上面方法計(jì)算出的橢圓如下:
可以看出結(jié)果非常的好。尤其是旋轉(zhuǎn)角度,計(jì)算的非常準(zhǔn)確。
下面是我的測試代碼,供參考。用到了些 Qt 的功能。
#include
#include
#include
#include
#include "picturebox.h"
#include
QImage threshold(const QImage &image, quint8 th)
{
int height = image.height();
int width = image.width();
QImage ret(width, height, QImage::Format_Indexed8);
ret.setColorCount(256);
for(int i = 0; i < 256; i++)
{
ret.setColor(i, qRgb(i, i, i));
}
for(int i = 0; i < height; i ++)
{
const uchar *pSrc = (uchar *)image.constScanLine(i);
uchar *pDest = (uchar *)ret.scanLine(i);
for( int j = 0; j < width; j ++)
{
pDest[j] = (pSrc[j] > th)? 255: 0;
}
}
return ret;
}
QImage toGray( const QImage &image )
{
int height = image.height();
int width = image.width();
QImage ret(width, height, QImage::Format_Indexed8);
ret.setColorCount(256);
for(int i = 0; i < 256; i++)
{
ret.setColor(i, qRgb(i, i, i));
}
qDebug () << image.format();
switch(image.format())
{
case QImage::Format_Indexed8:
case QImage::Format_Grayscale8:
for(int i = 0; i < height; i ++)
{
const uchar *pSrc = (uchar *)image.constScanLine(i);
uchar *pDest = (uchar *)ret.scanLine(i);
memcpy(pDest, pSrc, width);
}
break;
case QImage::Format_RGB32:
case QImage::Format_ARGB32:
case QImage::Format_ARGB32_PremulTIplied:
for(int i = 0; i < height; i ++)
{
const QRgb *pSrc = (QRgb *)image.constScanLine(i);
uchar *pDest = (uchar *)ret.scanLine(i);
for( int j = 0; j < width; j ++)
{
pDest[j] = qGray(pSrc[j]);
}
}
break;
}
return ret;
}
QPointF center(const QImage &image, int value)
{
if(image.isNull() || image.format() != QImage::Format_Indexed8)
{
return QPointF(-1, -1);
}
int width = image.width();
int height = image.height();
int x_mean = 0;
int y_mean = 0;
int count = 0;
for(int j = 0; j < height; j ++)
{
const uchar * p = image.constScanLine(j);
for(int i = 0; i < width; i++)
{
if( p[i] == value )
{
x_mean += i;
y_mean += j;
count++;
}
}
}
return QPointF((double)x_mean / count, (double)y_mean / count);
}
struct ELLIPSE_PARA
{
double x_mean; //橢圓的中心坐標(biāo) x
double y_mean; //橢圓的中心坐標(biāo) y
double r1; //橢圓的長軸半徑
double r2; //橢圓的短軸半徑
double theta; //橢圓的長軸與 x 軸的夾角(逆時(shí)針)
};
/**
* @brief ellipseFit 將一片區(qū)域當(dāng)作橢圓來估計(jì)五個(gè)幾何參數(shù)
* @param image
* @param value
* @param para
*/
bool ellipseFit(const QImage &image, int value, ELLIPSE_PARA * para)
{
if(image.isNull() || image.format() != QImage::Format_Indexed8)
{
return false;
}
QPointF c = center(image, value);
int width = image.width();
int height = image.height();
double n01 = c.x();
double n10 = c.y();
double mu20 = 0.0;
double mu02 = 0.0;
double mu11 = 0.0;
int count = 0;
for(int row = 0; row < height; row ++)
{
const uchar * p = image.constScanLine(row);
for(int col = 0; col < width; col++)
{
if( p[col] == value )
{
mu02 += (col - n01) * (col - n01);
mu20 += (row - n10) * (row - n10);
mu11 += (col - n01) * (row - n10);
count ++;
}
}
}
if(count == 0)
{
return false;
}
mu20 /= count;
mu02 /= count;
mu11 /= count;
double t1 = mu20 + mu02;
double t2 = mu20 - mu02;
double t3 = sqrt(t2 * t2 + 4 * mu11 * mu11);
double r1 = sqrt(2 * ( t1 + t3) );
double r2 = sqrt(2 * ( t1 - t3) );
double theta = - atan2(2 * mu11, mu02 - mu20) / 2.0;
para->r1 = r1;
para->r2 = r2;
para->theta = theta;
para->x_mean = n01;
para->y_mean = n10;
return true;
}
int main(int argc, char *argv[])
{
QApplicaTIon a(argc, argv);
QImage image("D:/test55.png");
QImage imageGray = toGray(image);
//imageGray = threshold(imageGray, 128);
ELLIPSE_PARA para;
ellipseFit(imageGray, 0, ¶);
qDebug() << para.r1;
qDebug() << para.r2;
qDebug() << para.theta * 180 / 3.14159;
QPointF c(para.x_mean, para.y_mean);
qDebug() << c;
QPainter painter(&image);
painter.setPen(Qt::red);
painter.translate(c);
painter.rotate(-para.theta * 180 / 3.14159);
painter.drawEllipse(QPointF(0, 0), para.r1, para.r2 );
PictureBox box;
box.seTImage(image);
box.show();
return a.exec();
}