Merge pull request #795 from edgarriba/master

add kornia to dockerfile

Author: rosbo

Dockerfile

@@ -328,6 +328,7 @@ RUN pip install bcolz && \
     pip install notebook==5.5.0 && \
     pip install olefile && \
     pip install opencv-python && \
+    pip install kornia && \
     pip install pandas_summary && \
     pip install pandocfilters && \
     pip install pexpect && \

tests/test_kornia.py

@@ -0,0 +1,19 @@
+import unittest
+
+import cv2
+import torch
+import kornia
+
+class TestKornia(unittest.TestCase):
+    def test_imread_opencv(self):
+        img = cv2.imread('/input/tests/data/dot.png')
+        img_t = kornia.image_to_tensor(img)
+
+        self.assertEqual(img.shape, (1, 1, 3))
+        self.assertEqual(img_t.shape, (3, 1, 1))
+
+    def test_grayscale_torch(self):
+        img_rgb = torch.rand(2, 3, 4, 5)
+        img_gray = kornia.rgb_to_grayscale(img_rgb)
+
+        self.assertEqual(img_gray.shape, (2, 1, 4, 5))