mirror of
https://github.com/opencv/opencv.git
synced 2024-11-27 04:36:36 +08:00
f40725bb50
Evangelidis, G.D. and Psarakis E.Z. "Parametric Image Alignment using Enhanced Correlation Coefficient Maximization", IEEE Transactions on PAMI, vol. 32, no. 10, 2008
75 lines
2.5 KiB
C++
75 lines
2.5 KiB
C++
#include "perf_precomp.hpp"
|
|
|
|
using namespace std;
|
|
using namespace cv;
|
|
using namespace perf;
|
|
using std::tr1::make_tuple;
|
|
using std::tr1::get;
|
|
|
|
CV_ENUM(MotionType, MOTION_TRANSLATION, MOTION_EUCLIDEAN, MOTION_AFFINE, MOTION_HOMOGRAPHY)
|
|
|
|
typedef std::tr1::tuple<MotionType> MotionType_t;
|
|
typedef perf::TestBaseWithParam<MotionType_t> TransformationType;
|
|
|
|
|
|
PERF_TEST_P(TransformationType, findTransformECC, /*testing::ValuesIn(MotionType::all())*/
|
|
testing::Values((int) MOTION_TRANSLATION, (int) MOTION_EUCLIDEAN,
|
|
(int) MOTION_AFFINE, (int) MOTION_HOMOGRAPHY)
|
|
)
|
|
{
|
|
|
|
Mat inputImage = imread(getDataPath("cv/shared/fruits.png"),0);
|
|
Mat img;
|
|
resize(inputImage, img, Size(216,216));
|
|
Mat templateImage;
|
|
|
|
int transform_type = get<0>(GetParam());
|
|
|
|
Mat warpMat;
|
|
Mat warpGround;
|
|
|
|
double angle;
|
|
switch (transform_type) {
|
|
case MOTION_TRANSLATION:
|
|
warpGround = (Mat_<float>(2,3) << 1.f, 0.f, 7.234f,
|
|
0.f, 1.f, 11.839f);
|
|
|
|
warpAffine(img, templateImage, warpGround,
|
|
Size(200,200), CV_INTER_LINEAR+CV_WARP_FILL_OUTLIERS+CV_WARP_INVERSE_MAP);
|
|
break;
|
|
case MOTION_EUCLIDEAN:
|
|
angle = CV_PI/30;
|
|
|
|
warpGround = (Mat_<float>(2,3) << (float)cos(angle), (float)-sin(angle), 12.123f,
|
|
(float)sin(angle), (float)cos(angle), 14.789f);
|
|
warpAffine(img, templateImage, warpGround,
|
|
Size(200,200), CV_INTER_LINEAR+CV_WARP_FILL_OUTLIERS+CV_WARP_INVERSE_MAP);
|
|
break;
|
|
case MOTION_AFFINE:
|
|
warpGround = (Mat_<float>(2,3) << 0.98f, 0.03f, 15.523f,
|
|
-0.02f, 0.95f, 10.456f);
|
|
warpAffine(img, templateImage, warpGround,
|
|
Size(200,200), CV_INTER_LINEAR+CV_WARP_FILL_OUTLIERS+CV_WARP_INVERSE_MAP);
|
|
break;
|
|
case MOTION_HOMOGRAPHY:
|
|
warpGround = (Mat_<float>(3,3) << 0.98f, 0.03f, 15.523f,
|
|
-0.02f, 0.95f, 10.456f,
|
|
0.0002f, 0.0003f, 1.f);
|
|
warpPerspective(img, templateImage, warpGround,
|
|
Size(200,200), CV_INTER_LINEAR+CV_WARP_FILL_OUTLIERS+CV_WARP_INVERSE_MAP);
|
|
break;
|
|
}
|
|
|
|
TEST_CYCLE()
|
|
{
|
|
if (transform_type<3)
|
|
warpMat = Mat::eye(2,3, CV_32F);
|
|
else
|
|
warpMat = Mat::eye(3,3, CV_32F);
|
|
|
|
findTransformECC(templateImage, img, warpMat, transform_type,
|
|
TermCriteria(TermCriteria::COUNT+TermCriteria::EPS, 5, -1));
|
|
}
|
|
SANITY_CHECK(warpMat, 1e-3);
|
|
}
|