remove opencv ad dep

openAGI · Mar 5, 2025 · cade14a · cade14a
1 parent 1712e66
commit cade14a
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/datum/utils/common_utils.py b/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
 
 

diff --git a/datum/version.py b/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'
diff --git a/requirements.txt b/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