fix: 🐛 remove imgaug for compat with numpy 2.x

Author: chaofengc

pyiqa/data/transforms.py

@@ -4,7 +4,6 @@
 from typing import Union
 from PIL import Image
 from collections.abc import Sequence
-from imgaug import augmenters as iaa
 import numpy as np
 
 import torch
@@ -36,8 +35,6 @@ def transform_mapping(key, args):
         return [PairedRandomRot90(args)]
     elif key == 'randomerase':
         return [PairedRandomErasing(**args)]
-    elif key == 'changecolor':
-        return [ChangeColorSpace(args)]
     elif key == 'totensor' and args:
         return [PairedToTensor()]
     else:
@@ -66,22 +63,6 @@ def __call__(self, imgs):
             return self.to_tensor(imgs) 
 
 
-class ChangeColorSpace:
-    """Pair version of center crop"""
-    def __init__(self, to_colorspace):
-        self.aug_op = iaa.color.ChangeColorspace(to_colorspace)
-
-    def __call__(self, imgs):
-        if _is_pair(imgs):
-            for i in range(len(imgs)):
-                tmpimg = self.aug_op.augment_image(np.array(imgs[i]))
-                imgs[i] = Image.fromarray(tmpimg)
-            return imgs 
-        else:
-            imgs = self.aug_op.augment_image(np.array(imgs))
-            return Image.fromarray(imgs)
-
-
 class PairedCenterCrop(tf.CenterCrop):
     """Pair version of center crop"""
     def forward(self, imgs):

requirements.txt

@@ -16,7 +16,6 @@ torchvision>=0.13
 tqdm
 yapf
 einops
-imgaug
 openai-clip
 facexlib
 transformers==4.37.2