init

requirements.txt

@@ -1,34 +1,13 @@
-apex==0.1
-av==11.0.0
-decord==0.6.0
-deepspeed==0.13.1
-einops==0.7.0
-ftfy==6.1.3
-fvcore==0.1.5.post20221221
-imageio==2.33.1
-lm_eval==0.4.1
-numpy==1.26.4
-omegaconf==2.3.0
-opencv_python==4.8.1.78
-packaging==24.0
-pandas==2.2.1
-Pillow==10.1.0
-pytest==8.1.1
-PyYAML==6.0.1
-regex==2023.10.3
-Requests==2.31.0
-scipy==1.12.0
-setuptools==68.2.2
-skimage==0.0
-submitit==1.5.1
-tensorboardX==2.6.2.2
-tensorflow==2.16.1
-termcolor==2.4.0
-timm==0.4.12
-torch==2.1.1
-torchvision==0.16.1
-tqdm==4.66.1
-transformers==4.36.1
-wandb==0.16.2
-wheel==0.42.0
-xformers==0.0.24
+decord
+einops
+lm_eval
+numpy
+packaging
+pytest
+setuptools
+skimage
+timm
+torch
+torchvision
+transformers
+wheel

transforms.py

@@ -410,34 +410,3 @@ class IdentityTransform(object):
 
     def __call__(self, data):
         return data
-
-
-if __name__ == "__main__":
-    trans = torchvision.transforms.Compose([
-        GroupScale(256),
-        GroupRandomCrop(224),
-        Stack(),
-        ToTorchFormatTensor(),
-        GroupNormalize(
-            mean=[.485, .456, .406],
-            std=[.229, .224, .225]
-        )]
-    )
-
-    im = Image.open('../tensorflow-model-zoo.torch/lena_299.png')
-
-    color_group = [im] * 3
-    rst = trans(color_group)
-
-    gray_group = [im.convert('L')] * 9
-    gray_rst = trans(gray_group)
-
-    trans2 = torchvision.transforms.Compose([
-        GroupRandomSizedCrop(256),
-        Stack(),
-        ToTorchFormatTensor(),
-        GroupNormalize(
-            mean=[.485, .456, .406],
-            std=[.229, .224, .225])
-    ])
-    print(trans2(color_group))