Perform action in BGRA and then reduce back to BGA for export
This commit is contained in:
parent
370e76aaaa
commit
0af8b43606
|
@ -12,8 +12,10 @@ username = os.getlogin()
|
|||
|
||||
#outer template
|
||||
outer = cv2.imread(f"/home/{username}/Desktop/running/shot1-scans/image-2-normal.png")
|
||||
outer = cv2.cvtColor(outer, cv2.COLOR_BGR2BGRA)
|
||||
#inner image
|
||||
inner = cv2.imread(f"/home/{username}/Desktop/frame.jpg")
|
||||
inner = cv2.cvtColor(inner, cv2.COLOR_BGR2BGRA)
|
||||
|
||||
rows, cols, ch = outer.shape
|
||||
ir, ic, ich = inner.shape
|
||||
|
@ -31,14 +33,19 @@ pts2 = np.float32([[0, 0], [ic, 0], [0, ir]])
|
|||
M = cv2.getAffineTransform(pts2, pts1)
|
||||
|
||||
# apply
|
||||
dst = cv2.warpAffine(inner, M, (cols, rows), borderMode=cv2.BORDER_TRANSPARENT)
|
||||
dst = cv2.warpAffine(inner, M, (cols, rows),
|
||||
flags = cv2.INTER_CUBIC,
|
||||
borderMode = cv2.BORDER_CONSTANT,
|
||||
borderValue = [0, 0, 0, 0]
|
||||
)
|
||||
|
||||
lower = np.array([0,0,0])
|
||||
upper= np.array([0,0,0])
|
||||
lower = np.array([0,0,0,0])
|
||||
upper= np.array([0,0,0,0])
|
||||
|
||||
# Create a mask. Threshold the HSV image to get only yellow colors
|
||||
mask = cv2.inRange(dst, lower, upper)
|
||||
mask = cv2.bitwise_not(mask)
|
||||
|
||||
final = combine_images(outer, dst, mask)
|
||||
final = cv2.cvtColor(final, cv2.COLOR_BGRA2BGR)
|
||||
cv2.imwrite("affine_poc3.png", final)
|
Loading…
Reference in New Issue