Create contourfinder.py

OpenCV/contourfinder.py

@@ -0,0 +1,121 @@
+import cv2 
+import sys
+from cv2 import CONTOURS_MATCH_I2
+import numpy as np
+
+#Takes two inputs as images
+#Original 1 and 2 are dummy data
+
+original =cv2.imread(r"Animate\images\flag (1).png")
+original2 =cv2.imread(r"Animate\images\brain2.png")
+
+# get key positions at which frame needs to be generated
+def list_of_positions(num_contours,num_frames=100):
+   positions=[]
+   for i in range(0, num_frames):
+      positions.append(int(num_contours/num_frames*i)) 
+   return positions 
+
+
+
+def contourfinder(image1, image2, text=None):
+    #Create two blank pages to write into
+    blank = np.zeros(image1.shape, dtype='uint8')
+    blank2 = np.zeros(image1.shape, dtype='uint8')
+
+
+    #Threshold and contours for image 1 and 2
+    threshold=cv2.Canny(image=image1, threshold1=100, threshold2=200)
+    contours, hierarchies = cv2.findContours(threshold, cv2.RETR_LIST, cv2.CHAIN_APPROX_SIMPLE)
+
+    threshold2=cv2.Canny(image=original2, threshold1=100, threshold2=200)
+    contours2, hierarchies2 = cv2.findContours(threshold2, cv2.RETR_LIST, cv2.CHAIN_APPROX_SIMPLE)
+
+
+    #Initialize three empty videos
+    vid1 = cv2.VideoWriter('vid1.mp4',cv2.VideoWriter_fourcc(*'mp4v'), 24, threshold.shape)
+    vid2 = cv2.VideoWriter('vid2.mp4',cv2.VideoWriter_fourcc(*'mp4v'), 24, threshold.shape) 
+    text_vid = cv2.VideoWriter('text_vid.mp4',cv2.VideoWriter_fourcc(*'mp4v'), 10, threshold.shape) 
+
+    
+    #Get positions
+    positions=list_of_positions((len(contours)))
+    frames=[]
+
+
+    #Loop over contours adding them to blank image then writing to video
+    for i in range(0, len(contours)):
+        cv2.drawContours(blank, contours=contours, contourIdx=i, color=(125, 200, 255), thickness=1) 
+
+        if i in positions:
+            frames.append(blank)
+            
+            #Complile to video
+            vid1.write(blank)
+        vid1.release()
+
+    positions=list_of_positions((len(contours2)))
+
+    for i in range(0, len(contours2)):
+        cv2.drawContours(blank2, contours=contours2, contourIdx=i, color=(125, 200, 255), thickness=1) 
+        if i in positions:
+            frames.append(blank2)
+            
+            vid2.write(blank2)
+        vid2.release()
+
+    
+    #Next is the text vid
+
+    if text !=None:	
+        # Reading an image in default mode
+        image = np.zeros(original.shape, dtype='uint8')
+            
+        # font
+        font = cv2.FONT_HERSHEY_COMPLEX
+
+        # org
+        org = (10, 400)
+
+        # fontScale
+        fontScale = 3
+
+        # Blue color in BGR
+        color = (186, 184, 108)
+
+        # Line thickness of 2 px
+        thickness = 4
+
+
+
+
+        def text_frames(text,image,org):
+            spacing=55 #spacing between letters
+            blink=image
+            cv2.imwrite(f"blink.png", blink)
+            for i in range(0, len(text)-1):
+                
+                text_vid.write(blink)
+            
+                # Using cv2.putText() method
+                image = cv2.putText(image, text[i], org, font,
+                    fontScale, color, thickness, cv2.LINE_AA)
+
+                #Take care of org spacing
+                org=(org[0]+spacing,org[1])
+                if text[i].isupper():
+                    org=(org[0] +spacing +1, org[1])
+                    print(f"Upper {text[i]}")
+                print(org)
+
+                # Displaying the image
+                cv2.imwrite(f"text_im{i}.png", image)
+
+                #Complile to video
+                text_vid.write(image)
+            text_vid.release()
+
+        text_frames(text,image,org)
+        
+
+contourfinder(original, original2, "spies gui Fly")