opencv - Python中的OpenCV 2.4估计Affine3D

Question

我正在尝试使用方法 cv2.estimateAffine3D 但没有成功。这是我的代码示例：

import numpy as np
import cv2

shape = (1, 4, 3)
source = np.zeros(shape, np.float32)

# [x, y, z]
source[0][0] = [857, 120, 854]
source[0][1] = [254, 120, 855]
source[0][2] = [256, 120, 255]
source[0][3] = [858, 120, 255]
target = source * 10

retval, M, inliers = cv2.estimateAffine3D(source, target)

当我尝试运行此示例时，我收到与此处的其他帖子相同的错误。

我正在使用 OpenCV 2.4.3 和 Python 2.7.3

请帮我！

Answer 1

这是一个已知的错误，已在2.4.4.

http://code.opencv.org/issues/2375

如果您只需要刚性（旋转+平移）对齐，这是标准方法：

def get_rigid(src, dst): # Assumes both or Nx3 matrices
    src_mean = src.mean(0)
    dst_mean = dst.mean(0)
    # Compute covariance 
    H = reduce(lambda s, (a,b) : s + np.outer(a, b), zip(src - src_mean, dst - dst_mean), np.zeros((3,3)))
    u, s, v = np.linalg.svd(H)
    R = v.T.dot(u.T) # Rotation
    T = - R.dot(src_mean) + dst_mean # Translation
    return np.hstack((R, T[:, np.newaxis]))