def test__binning_y(self):
cm = PinholeCameraModel.from_yaml_file(kinect_v2_camera_info())
cm.roi = [0, 0, 100, 100]
cm.target_size = (640, 480)
cm.binning_y = 0.5
self.assertEqual(cm.target_size, (640, 200))
cm.target_size = (640, 480)
cm.roi = [0, 0, 97, 100]
cm.binning_y = 2
self.assertEqual(cm.target_size, (640, 48))
def test_rectify_point(self):
cm = PinholeCameraModel.from_yaml_file(kinect_v2_camera_info())
testing.assert_equal(cm.rectify_point([0, 0]).shape, (2,))
testing.assert_equal(cm.rectify_point((0, 0)).shape, (2,))
testing.assert_equal(cm.rectify_point(np.array((0, 0))).shape, (2,))
testing.assert_equal(cm.rectify_point([[0, 0]]).shape, (1, 2))
testing.assert_equal(
cm.rectify_point(np.array([[0, 0]])).shape, (1, 2))
testing.assert_equal(
cm.rectify_point(np.zeros((10, 2))).shape, (10, 2))
def test_points_roi(self):
cm = PinholeCameraModel.from_yaml_file(kinect_v2_camera_info())
points = [[874.5, 680],
[875, 680],
[875, 679.5],
[1072, 680],
[1072.5, 680],
[1072, 859],
[1072, 869.5]]
testing.assert_equal(
cm.points_in_roi(points),
[False, True, False, True, False, True, False])
def test__target_size(self):
cm = PinholeCameraModel.from_yaml_file(kinect_v2_camera_info())
org_roi = copy.deepcopy(cm.roi)
cm.target_size = (640, 480)
self.assertEqual(cm.target_size, (640, 480))
resize_K = cm.K.copy()
resize_P = cm.P.copy()
cm.roi = [0, 0, 100, 100]
cm.target_size = (640, 480)
cm.roi = org_roi
testing.assert_equal(resize_K, cm.K)
testing.assert_equal(resize_P, cm.P)
def test__roi(self):
cm = PinholeCameraModel.from_yaml_file(kinect_v2_camera_info())
org_roi = copy.deepcopy(cm.roi)
org_K = cm.K.copy()
org_P = cm.P.copy()
cm.roi = [0, 0, 100, 100]
testing.assert_almost_equal(org_K[:3, :2], cm.K[:3, :2])
testing.assert_almost_equal(cm.K[:2, 2], [951.8467, 506.9212],
decimal=3)
testing.assert_almost_equal(org_P[:3, :2], cm.P[:3, :2])
testing.assert_almost_equal(cm.P[:2, 2], [951.8467, 506.9212],
decimal=3)
cm.roi = org_roi
testing.assert_almost_equal(org_K, cm.K)
testing.assert_almost_equal(org_P, cm.P)
def test_draw_roi(self):
cm = PinholeCameraModel.from_yaml_file(kinect_v2_camera_info())
img_org = kinect_v2_image()
img = img_org.copy()
cm.draw_roi(img, copy=True)
testing.assert_equal(img, img_org)
pil_img = Image.fromarray(img_org)
gray_pil_img = pil_img.convert("L")
gray_img_org = np.array(gray_pil_img, dtype=np.uint8)
gray_img = gray_img_org.copy()
cm.draw_roi(gray_img, copy=False)
testing.assert_equal(gray_img, gray_img_org)
cm.draw_roi(gray_img, copy=True) # ignore copy=True
testing.assert_equal(gray_img, gray_img_org)
alpha_pil_img = pil_img.convert("RGBA")
alpha_img_org = np.array(alpha_pil_img, dtype=np.uint8)
alpha_img = alpha_img_org.copy()
cm.draw_roi(alpha_img, copy=True)
testing.assert_equal(alpha_img, alpha_img_org)
def test_kinect_v2_camera_info(self):
info_path = kinect_v2_camera_info()
PinholeCameraModel.from_yaml_file(info_path)
def setUpClass(cls):
cls.cm = PinholeCameraModel.from_yaml_file(kinect_v2_camera_info())
def test_rectify_image(self):
cm = PinholeCameraModel.from_yaml_file(kinect_v2_camera_info())
img = kinect_v2_image()
cm.rectify_image(img)