如何用Python中的矩阵中的变量求解方程?

时间:2016-02-04 14:50:11

标签: python numpy matrix

我在Pyhon编码,我正在研究立体相关。我想解决这个等式:m = K.T.M

是的,K,M是知道的。

其中:

M 是笛卡尔坐标系中一个点的齐次坐标" world"

M=np.array([X,Y,Z,1])

K 是左相机的内在矩阵

K=np.matrix([ [fx,  0, cx, 0],
              [ 0, fy, cy, 0], 
              [ 0,  0,  1, 0]])

m 左侧相机的M点视图

m=np.array([x,y,1])

T 是转移到" world"坐标系到左摄像机坐标系。

T= np.matrix([[x00, x01, x02, Tx],
              [x10, x11, x12, Ty], 
              [x20, x21, x22, Tz], 
              [0  , 0  , 0  , 1 ]])

所以我想解决这个等式来找到 T 但是如果不给变量赋值就不可能创建矩阵。

有人有解决方案吗?

由于 最好的问候

1 个答案:

答案 0 :(得分:2)

如果您需要通用解决方案,可以使用Sympy,它允许您使用符号表达式。在以下代码中,表达式K.T.M = m被重新表述为标准线性方程HH.xx = mm,其中xx是从T提取未知数的向量:

from IPython.display import display
import sympy as sy

sy.init_printing()  # LaTeX like pretty printing for IPython

# declaring symbolic variables:
x, y, X, Y, Z, fx, fy, cx, cy = sy.symbols("x y X Y Z f_x f_y c_x c_y", real=True)
x00, x01, x02, x10, x11 = sy.symbols("x00, x01, x02, x10, x11", real=True)
x12, x20, x21, x22 = sy.symbols("x12, x20, x21, x22", real=True)
Tx, Ty, Tz = sy.symbols(" T_x T_y T_z", real=True)

# Building matrices and vectors:
M = sy.Matrix([X, Y, Z, 1])
m = sy.Matrix([x, y, 1])
K = sy.Matrix([[fx,  0, cx, 0],
               [0,  fy, cy, 0],
               [0,   0,  0, 1]])
T = sy.Matrix([[x00, x01, x02, Tx],
               [x10, x11, x12, Ty],
               [x20, x21, x22, Tz],
               [0,     0,   0,  1]])

print("KTM = K.T.M = ")
KTM = sy.simplify(K*T*M)
display(KTM)

print("Vector of Unkowns xx.T = ")
xx = sy.Matrix(list(T.atoms(sy.Symbol)))
display(xx.T)
print("For equation HH.xx = mm, HH = ")
HH = KTM[:2, :].jacobian(xx)  # calculate the derivative for each unknown
display(HH)

正如@ Sven-Marnach已经指出的那样,没有足够的方程式来获得独特的解决方案。由于向量KTMm的最后一行是1,因此十二个变量只有两个方程式。

如果您有多个像素评估,即多对(m, M),您可以使用Numpy's Least Squares Solver来查找解决方案。