more coherent version testing

spmallick · Jul 31, 2024 · 218cf76 · 218cf76
1 parent 824feb1
commit 218cf76
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/VideoStabilization/video_stabilization.py b/VideoStabilization/video_stabilization.py
@@ -91,7 +91,7 @@ def fixBorder(frame):
   curr_pts = curr_pts[idx]
 
   #Find transformation matrix
-  if float((cv2.__version__)[0:3])<=3.0:
+  if int((cv2.__version__).split('.')[0]) <= 3:
      m = cv2.estimateRigidTransform(prev_pts, curr_pts, fullAffine=False) #will only work with OpenCV-3 or less
 
   else: