animation/fourcell/notes/affine_poc2.py

33 lines
781 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.png", final)