From 0af8b43606d78d2fd405eebfe58ce8b6dad5063c Mon Sep 17 00:00:00 2001
From: mattmcw <matt@sixteenmillimeter.com>
Date: Mon, 20 Feb 2023 18:41:11 -0500
Subject: [PATCH] Perform action in BGRA and then reduce back to BGA for export

---
 fourcell/notes/affine_poc3.py | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/fourcell/notes/affine_poc3.py b/fourcell/notes/affine_poc3.py
index c863c49..88701bd 100644
--- a/fourcell/notes/affine_poc3.py
+++ b/fourcell/notes/affine_poc3.py
@@ -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)
\ No newline at end of file