Update video_stabilization.py

For OpenCV version greater than 3.0 assigned "estimateAffine2D" function for finding transformation matrix.

Author: pranav-gupta-7

VideoStabilization/video_stabilization.py

@@ -91,7 +91,11 @@ def fixBorder(frame):
   curr_pts = curr_pts[idx]
 
   #Find transformation matrix
-  m = cv2.estimateRigidTransform(prev_pts, curr_pts, fullAffine=False) #will only work with OpenCV-3 or less
+  if float((cv2.__version__)[0:3])<=3.0:
+     m = cv2.estimateRigidTransform(prev_pts, curr_pts, fullAffine=False) #will only work with OpenCV-3 or less
+   
+  else:
+    m,_ = cv2.estimateAffine2D(prev_pts, curr_pts)# will work with OpenCV>3.0
 
   # Extract traslation
   dx = m[0,2]
@@ -165,4 +169,4 @@ def fixBorder(frame):
 cap.release()
 out.release()
 # Close windows
-cv2.destroyAllWindows()
+cv2.destroyAllWindows()