mirror of
https://github.com/opencv/opencv.git
synced 2024-12-15 18:09:11 +08:00
108 lines
3.6 KiB
C++
108 lines
3.6 KiB
C++
|
// This file is part of OpenCV project.
|
||
|
// It is subject to the license terms in the LICENSE file found in the top-level directory
|
||
|
// of this distribution and at http://opencv.org/license.html
|
||
|
|
||
|
// This code is also subject to the license terms in the LICENSE_WillowGarage.md file found in this module's directory
|
||
|
|
||
|
#include "test_precomp.hpp"
|
||
|
|
||
|
namespace opencv_test { namespace {
|
||
|
|
||
|
class RgbdDepthRegistrationTest
|
||
|
{
|
||
|
public:
|
||
|
RgbdDepthRegistrationTest() { }
|
||
|
~RgbdDepthRegistrationTest() { }
|
||
|
|
||
|
void run()
|
||
|
{
|
||
|
// Test all three input types for no-op registrations (where a depth image is registered to itself)
|
||
|
noOpRandomRegistrationTest<unsigned short>(100, 2500);
|
||
|
noOpRandomRegistrationTest<float>(0.1f, 2.5f);
|
||
|
noOpRandomRegistrationTest<double>(0.1, 2.5);
|
||
|
|
||
|
// Test sentinel value handling, occlusion, and dilation
|
||
|
{
|
||
|
// K from a VGA Kinect
|
||
|
Mat K = (Mat_<float>(3, 3) << 525., 0., 319.5, 0., 525., 239.5, 0., 0., 1.);
|
||
|
|
||
|
int width = 640, height = 480;
|
||
|
|
||
|
// All elements are zero except for first two along the diagonal
|
||
|
Mat_<unsigned short> vgaDepth(height, width, (unsigned short)0);
|
||
|
vgaDepth(0, 0) = 1001;
|
||
|
vgaDepth(1, 1) = 1000;
|
||
|
|
||
|
Mat_<unsigned short> registeredDepth;
|
||
|
registerDepth(K, K, Mat(), Matx44f::eye(), vgaDepth, Size(width, height), registeredDepth, true);
|
||
|
|
||
|
// We expect the closer depth of 1000 to occlude the more distant depth and occupy the
|
||
|
// upper four left pixels in the depth image because of dilation
|
||
|
Mat_<unsigned short> expectedResult(height, width, (unsigned short)0);
|
||
|
expectedResult(0, 0) = 1000;
|
||
|
expectedResult(0, 1) = 1000;
|
||
|
expectedResult(1, 0) = 1000;
|
||
|
expectedResult(1, 1) = 1000;
|
||
|
|
||
|
Mat ad;
|
||
|
absdiff(registeredDepth, expectedResult, ad);
|
||
|
ASSERT_GT(std::numeric_limits<double>::min(), abs(sum(ad)[0])) << "Dilation and occlusion";
|
||
|
}
|
||
|
}
|
||
|
|
||
|
template <class DepthDepth>
|
||
|
void noOpRandomRegistrationTest(DepthDepth minDepth, DepthDepth maxDepth)
|
||
|
{
|
||
|
// K from a VGA Kinect
|
||
|
Mat K = (Mat_<float>(3, 3) << 525., 0., 319.5, 0., 525., 239.5, 0., 0., 1.);
|
||
|
|
||
|
// Create a random depth image
|
||
|
RNG rng;
|
||
|
Mat_<DepthDepth> randomVGADepth(480, 640);
|
||
|
rng.fill(randomVGADepth, RNG::UNIFORM, minDepth, maxDepth);
|
||
|
|
||
|
Mat registeredDepth;
|
||
|
registerDepth(K, K, Mat(), Matx44f::eye(), randomVGADepth, Size(640, 480), registeredDepth);
|
||
|
|
||
|
// Check per-pixel relative difference
|
||
|
|
||
|
Mat ad;
|
||
|
absdiff(registeredDepth, randomVGADepth, ad);
|
||
|
Mat mmin;
|
||
|
cv::min(registeredDepth, randomVGADepth, mmin);
|
||
|
Mat rel = ad / mmin;
|
||
|
double maxDiff = cv::norm(rel, NORM_INF);
|
||
|
ASSERT_GT(1e-07, maxDiff) << "No-op registration";
|
||
|
}
|
||
|
};
|
||
|
|
||
|
TEST(RGBD_DepthRegistration, compute)
|
||
|
{
|
||
|
RgbdDepthRegistrationTest test;
|
||
|
test.run();
|
||
|
}
|
||
|
|
||
|
TEST(RGBD_DepthRegistration, issue_2234)
|
||
|
{
|
||
|
Matx33f intrinsicsDepth(100, 0, 50, 0, 100, 50, 0, 0, 1);
|
||
|
Matx33f intrinsicsColor(100, 0, 200, 0, 100, 50, 0, 0, 1);
|
||
|
|
||
|
Mat_<float> depthMat(100, 100, (float)0.);
|
||
|
for(int i = 1; i <= 100; i++)
|
||
|
{
|
||
|
for(int j = 1; j <= 100; j++)
|
||
|
depthMat(i-1,j-1) = (float)j;
|
||
|
}
|
||
|
|
||
|
Mat registeredDepth;
|
||
|
registerDepth(intrinsicsDepth, intrinsicsColor, Mat(), Matx44f::eye(), depthMat, Size(400, 100), registeredDepth);
|
||
|
|
||
|
Rect roi( 150, 0, 100, 100 );
|
||
|
Mat subM(registeredDepth,roi);
|
||
|
|
||
|
EXPECT_EQ(0, cvtest::norm(subM, depthMat, NORM_INF));
|
||
|
}
|
||
|
|
||
|
|
||
|
}} // namespace
|