Skip to content

Commit 30be198

Browse files
committed
tests for python bindings added
1 parent a5be051 commit 30be198

File tree

2 files changed

+49
-4
lines changed

2 files changed

+49
-4
lines changed
Lines changed: 45 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,45 @@
1+
#!/usr/bin/env python
2+
3+
# Python 2/3 compatibility
4+
from __future__ import print_function
5+
6+
import os, numpy
7+
8+
import cv2 as cv
9+
10+
from tests_common import NewOpenCVTests
11+
12+
class rgbd_test(NewOpenCVTests):
13+
14+
def test_computeRgbdPlane(self):
15+
16+
depth_image = self.get_sample('/cv/rgbd/depth.png', cv.IMREAD_ANYDEPTH)
17+
if depth_image is None:
18+
raise unittest.SkipTest("Missing files with test data")
19+
20+
K = numpy.array([[525, 0, 320.5], [0, 525, 240.5], [0, 0, 1]])
21+
points3d = cv.rgbd.depthTo3d(depth_image, K)
22+
normals_computer = normals_computer = cv.rgbd.RgbdNormals_create(480, 640, 5, K)
23+
normals = normals_computer.apply(points3d)
24+
rgbd_plane = cv.rgbd.RgbdPlane_create(cv.rgbd.RgbdPlane_RGBD_PLANE_METHOD_DEFAULT, 40, 1600, 0.01, 0, 0, 0)
25+
_, planes_coeff = rgbd_plane.apply(points3d, normals)
26+
27+
planes_coeff_expected = \
28+
numpy.asarray([[[-0.02447728, -0.8678335 , -0.49625182, 4.02800846]],
29+
[[-0.05055107, -0.86144137, -0.50533485, 3.95456314]],
30+
[[-0.03294908, -0.86964548, -0.49257591, 3.97052431]],
31+
[[-0.02886586, -0.87153459, -0.48948362, 7.77550507]],
32+
[[-0.04455929, -0.87659335, -0.47916424, 3.93200684]],
33+
[[-0.21514639, 0.18835169, -0.95824611, 7.59479475]],
34+
[[-0.01006953, -0.86679155, -0.49856904, 4.01355648]],
35+
[[-0.00876531, -0.87571168, -0.48275498, 3.96768975]],
36+
[[-0.06395926, -0.86951321, -0.48975089, 4.08618736]],
37+
[[-0.01403128, -0.87593341, -0.48222789, 7.74559402]],
38+
[[-0.01143177, -0.87495202, -0.4840748 , 7.75355816]]],
39+
dtype=numpy.float32)
40+
41+
eps = 0.05
42+
self.assertLessEqual(cv.norm(planes_coeff, planes_coeff_expected, cv.NORM_L2), eps)
43+
44+
if __name__ == '__main__':
45+
NewOpenCVTests.bootstrap()

modules/rgbd/src/odometry.cpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1010,8 +1010,8 @@ Ptr<DepthCleaner> DepthCleaner::create(int depth_in, int window_size_in, int met
10101010
}
10111011

10121012
RgbdPlane::RgbdPlane(int method, int block_size,
1013-
int min_size, double threshold, double sensor_error_a = 0,
1014-
double sensor_error_b = 0, double sensor_error_c = 0) :
1013+
int min_size, double threshold, double sensor_error_a,
1014+
double sensor_error_b, double sensor_error_c) :
10151015
method_(method),
10161016
block_size_(block_size),
10171017
min_size_(min_size),
@@ -1022,8 +1022,8 @@ RgbdPlane::RgbdPlane(int method, int block_size,
10221022
{}
10231023

10241024
Ptr<RgbdPlane> RgbdPlane::create(int method, int block_size, int min_size, double threshold,
1025-
double sensor_error_a = 0, double sensor_error_b = 0,
1026-
double sensor_error_c = 0) {
1025+
double sensor_error_a, double sensor_error_b,
1026+
double sensor_error_c ) {
10271027
return makePtr<RgbdPlane>(method, block_size, min_size, threshold,
10281028
sensor_error_a, sensor_error_b, sensor_error_c);
10291029
}

0 commit comments

Comments
 (0)