Zadanie 12

This commit is contained in:
Sebastian Spaloniak 2020-05-24 20:09:21 +02:00
parent 1139cf8f8b
commit 51b9d65213

View File

@ -25,8 +25,109 @@ PNM* CornerHarris::transform()
height = image->height(); height = image->height();
PNM* newImage = new PNM(width, height, QImage::Format_Mono); PNM* newImage = new PNM(width, height, QImage::Format_Mono);
math::matrix<double>
imx(width, height),
imy(width, height),
imxy(width, height),
corncan(width, height),
cornnonsup(width, height);
PNM* tempImage = ConversionGrayscale(image).transform();
qDebug() << Q_FUNC_INFO << "Not implemented yet!"; BlurGaussian blurGauss(tempImage);
blurGauss.setParameter("size", 3);
blurGauss.setParameter("sigma", 3.6);
tempImage = blurGauss.transform();
EdgeSobel edgsob(tempImage);
math::matrix<float>* xgra = edgsob.rawHorizontalDetection();
math::matrix<float>* ygra = edgsob.rawVerticalDetection();
for (int x = 0; x < width; x++)
{
for (int y = 0; y < height; y++)
{
double xgraphic = (*xgra)(x, y), ygraphic = (*ygra)(x, y);
imx(x, y) = xgraphic * xgraphic;
imy(x, y) = ygraphic * ygraphic;
imxy(x, y) = xgraphic * ygraphic;
}
}
for (int x = 0; x < width; x++)
{
for (int y = 0; y < height; y++)
{
if (x == 0 || x == width - 1 || y == 0 || y == height - 1)
{
corncan(x, y) = 0;
cornnonsup(x, y) = 0;
}
else
{
double sx = 0, sy = 0, sxy = 0;
for (int k = -1; k < 2; k++)
{
for (int j = -1; j < 2; j++)
{
double g_val = BlurGaussian::getGauss(k, j, sigma);
sx += imx(x + k, y + j)*g_val;
sy += imy(x + k, y + j)*g_val;
sxy += imxy(x + k, y + j)*g_val;
}
}
sx /= sigma_weight;
sy /= sigma_weight;
sxy /= sigma_weight;
math::matrix<double> H(2, 2);
H(0, 0) = sx; H(1, 1) = sy; H(0, 1) = sxy; H(1, 0) = sxy;
double r = H(0, 0)*H(1, 1) - H(0, 1)*H(1, 0) - k_param*pow(H(0, 0) + H(1, 1), 2);
corncan(x, y) = (r > threshold) ? r : 0;
}
}
}
bool maknonsup = true;
while (maknonsup)
{
maknonsup = false;
for (int x = 1; x < width - 1; x++)
{
for (int y = 1; y < height - 1; y++)
{
double val = corncan(x, y);
if (val > corncan(x - 1, y - 1) && val > corncan(x - 1, y) && val > corncan(x - 1, y + 1) && val > corncan(x, y - 1) && val > corncan(x, y + 1) && val > corncan(x + 1, y - 1) && val > corncan(x + 1, y) && val > corncan(x + 1, y + 1))
{
cornnonsup(x, y) = val;
}
else
{
if (val > 0)
{
maknonsup = true;
}
cornnonsup(x, y) = 0;
}
}
}
corncan = cornnonsup;
}
for (int x = 0; x < width; x++)
{
for (int y = 0; y < height; y++)
{
if (corncan(x, y) == 0){
newImage->setPixel(x, y, 0);
}
else
{
newImage->setPixel(x, y, 1);
}
}
}
return newImage; return newImage;
} }