Update contourfinder.py

OpenCV/contourfinder.py

@@ -18,7 +18,7 @@ def list_of_positions(num_contours,num_frames=100):
 
 
 
-def contourfinder(image1, image2, text=None):
+def contourfinder(image1, image2, text=None, num_frames=100):
     #Create two blank pages to write into
     blank = np.zeros(image1.shape, dtype='uint8')
     blank2 = np.zeros(image1.shape, dtype='uint8')
@@ -30,7 +30,7 @@ def contourfinder(image1, image2, text=None):
 
     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)
@@ -52,7 +52,8 @@ def contourfinder(image1, image2, text=None):
             
             #Complile to video
             vid1.write(blank)
-        vid1.release()
+    last_image1=blank
+    vid1.release()
 
     positions=list_of_positions((len(contours2)))
 
@@ -62,7 +63,8 @@ def contourfinder(image1, image2, text=None):
             frames.append(blank2)
             
             vid2.write(blank2)
-        vid2.release()
+    last_image2=blank2
+    vid2.release()
 
     
     #Next is the text vid
@@ -116,6 +118,7 @@ def contourfinder(image1, image2, text=None):
             text_vid.release()
 
         text_frames(text,image,org)
+    return last_image1,last_image2
         
 
-contourfinder(original, original2, "spies gui Fly")
+#contourfinder(original, original2, "spies gui Fly")