Update codes.py

codes.py

@@ -95,6 +95,25 @@ def workaround_matrix(Affine_mtrx0, acc = 2):
     Affine_mtrx_adj[:,:,2]*=acc
     return Affine_mtrx_adj
 
+def inv_AM(Affine_mtrx):
+    AM3 = mtrx3(Affine_mtrx)
+    AM_inv = torch.linalg.inv(AM3)
+    return AM_inv[:,0:2,:]
+
+def mtrx3(Affine_mtrx):
+    mtrx_shape = Affine_mtrx.shape
+    if len(mtrx_shape)==3:
+        N_Mbatches = mtrx_shape[0]
+        AM3 = torch.zeros( [N_Mbatches,3,3]).to(device)
+        AM3[:,0:2,:] = Affine_mtrx
+        AM3[:,2,2] = 1
+    elif len(mtrx_shape)==2:
+        N_Mbatches = 1
+        AM3 = torch.zeros([3,3]).to(device)
+        AM3[0:2,:] = Affine_mtrx
+        AM3[2,2] = 1 
+    return AM3
+
 def standarize_point(d, dim=128, flip = False):
     if flip:
         d = -d