Python中的OpenCV 2.4 estimateAffine3D

时间:2013-04-12 05:47:04

标签: opencv python-2.7 numpy

我正在尝试使用方法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)

当我尝试运行此示例时,我获得了与其他帖子here相同的错误。

我正在使用OpenCV 2.4.3和Python 2.7.3

请帮助我!

1 个答案:

答案 0 :(得分: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]))