requirements updated

.gitignore

@@ -121,4 +121,5 @@ results/
 *.jpg
 *.csv
 *.json
-*.png
+*.png
+*.zip

data/twitter/data_loader.py

@@ -1,7 +1,6 @@
 
-import cv2
-import pickle
 from PIL import Image
+from cv2 import cv2
 from numpy import asarray
 
 from data.data_loader import DatasetLoader

main.py

@@ -1,6 +1,5 @@
 import argparse
 
-from lime_main import lime_main
 from optuna_main import optuna_main
 from test_main import test_main
 from torch_main import torch_main
@@ -52,8 +51,6 @@ if __name__ == '__main__':
         optuna_main(config, args.use_optuna)
     elif args.just_test is not None:
         test_main(config, args.just_test)
-    elif args.use_lime is not None:
-        lime_main(config, args.use_lime)
     else:
         torch_main(config)
 

requirements.txt

@@ -5,7 +5,6 @@ Pillow==8.2.0
 pytorch-lightning==1.2.10
 pytorch-model-summary==0.1.2
 pytorchtools==0.0.2
-scikit-image==0.17.2
 scikit-learn==0.24.2
 scipy==1.5.4
 seaborn==0.11.1
@@ -14,4 +13,7 @@ torchmetrics==0.2.0
 torchsummary==1.5.1
 torchvision==0.9.1
 tqdm==4.60.0
-transformers==4.5.1
+tokenizers==0.13.3
+transformers==4.30.0
+timm==0.4.5
+albumentations==0.5.2