32 lines
782 B
Python
32 lines
782 B
Python
|
import cv2
|
||
|
|
import numpy as np
|
||
|
|
import os
|
||
|
|
|
||
|
|
username = os.getlogin()
|
||
|
|
|
||
|
|
#outer template
|
||
|
|
outer = cv2.imread(f"/home/{username}/Desktop/running/shot1-scans/image-2-normal.png")
|
||
|
|
#inner image
|
||
|
|
inner = cv2.imread(f"/home/{username}/Desktop/frame.jpg")
|
||
|
|
|
||
|
|
rows, cols, ch = outer.shape
|
||
|
|
ir, ic, ich = inner.shape
|
||
|
|
|
||
|
|
print(f'{cols}x{rows}')
|
||
|
|
print(f'{ic}x{ir}')
|
||
|
|
|
||
|
|
# destination ponts on outer image
|
||
|
|
pts1 = np.float32([[1445, 11429], [1445, 6832], [5009, 11429]])
|
||
|
|
|
||
|
|
# corresponding points on inner image
|
||
|
|
pts2 = np.float32([[0, 0], [ic, 0], [0, ir]])
|
||
|
|
|
||
|
|
# transform inner to points on outer
|
||
|
|
M = cv2.getAffineTransform(pts2, pts1)
|
||
|
|
|
||
|
|
# apply
|
||
|
|
dst = cv2.warpAffine(inner, M, (cols, rows), borderMode=cv2.BORDER_TRANSPARENT)
|
||
|
|
|
||
|
|
final = outer.copy()
|
||
|
|
final[0:rows, 0:cols] = dst
|
||
|
|
cv2.imwrite("affine_poc2-dst.png", dst)
|