Remove opencv as dep (#31)

* remove opencv ad dep

* fix test

* fix test

datum/utils/common_utils.py

@@ -266,10 +266,10 @@ def check_and_image_shape(item: ValueType, shape: list) -> list:
   if len(item.shape) > 0:
     item = str(item[0])
   if item.endswith(('.jpg', '.jpeg', '.png')):
-    import cv2
-    im = cv2.imread(item)
+    from PIL import Image
+    im = Image.open(item)
     if im is not None:
-      return list(im.shape)
+      return list(np.asarray(im).shape)
   return shape
 
 

datum/version.py

@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = '1.12.1'
+__version__ = '1.12.2'

requirements.txt

@@ -1,6 +1,6 @@
 # core lib dep
 tensorflow <2.19
-opencv-python==4.5.*
+pillow >=9.2.0
 absl-py>=0.10
 tqdm>=4.39.0
 wrapt>=1.14.1

tests/test_dataset.py

@@ -118,7 +118,7 @@ def test_train_fn(self):
     ds = self._dataset.train_fn('train', False)
     batch = next(iter(ds))
     self.assertEqual(batch['image'].shape, [1, 281, 500, 3])
-    self.assertEqual(batch['label'].shape, [1, 281, 500, 3])
+    self.assertEqual(batch['label'].shape, [1, 281, 500, 1])
 
 
 class TestTextJsonDataset(absltest.TestCase):

tests/test_parser.py

@@ -136,4 +136,4 @@ def test_parser(self):
     batch = next(iter(dataset))
     self.assertEqual(list(batch.keys()), ['image', 'label'])
     self.assertEqual(batch['image'].shape, [1, 366, 500, 3])
-    self.assertEqual(batch['label'].shape, [1, 366, 500, 3])
+    self.assertEqual(batch['label'].shape, [1, 366, 500, 1])