Merge pull request #852 from Kaggle/adding-pydegensac

Adding pydegensac

Author: rosbo

Dockerfile

@@ -440,6 +440,7 @@ RUN pip install flashtext && \
     pip install cloud-tpu-client && \
     pip install tensorflow-datasets && \
     pip install pydub && \
+    pip install pydegensac && \
     /tmp/clean-layer.sh
 
 # Tesseract and some associated utility packages

tests/test_pydegensac.py

@@ -0,0 +1,18 @@
+import unittest
+
+import pydegensac
+import numpy as np
+
+
+class TestPydegensac(unittest.TestCase):
+    def test_find_homography(self):
+        src_pts = np.float32([ [0,0],[0,1],[1,1],[1,0] ]).reshape(-1,2)
+        dst_pts = np.float32([ [0,0],[0,-1],[-1,-1],[-1,0] ]).reshape(-1,2)
+
+        H, mask = pydegensac.findHomography(src_pts, dst_pts, 4, 1)
+
+        self.assertEqual(3, len(H))
+        self.assertEqual(4, len(mask))
+
+
+