updated

requirements.txt

@@ -1,14 +1,23 @@
-albumentations==1.3.0
-keras==2.11.0
-matplotlib==3.6.2
-numpy==1.21.5
+# Base----------------------------------
+albumentations>=1.1.0
+matplotlib
+opencv-python>=4.1.1.26
 opencv-python-headless==4.5.4.60
-pandas==1.4.4
-Pillow==9.4.0
-pycocotools==2.0.6
-PyYAML==6.0
-scikit_image==0.19.3
-skimage==0.0
-streamlit==1.20.0
-torch==2.0.0+cpu
-torchvision==0.15.1+cpu
+Pillow
+PyYAML
+scikit-image
+scikit-learn
+scipy
+torch>=1.10.0
+torchvision>=0.11.0
+numpy
+protobuf<=3.20.1
+pandas
+
+# Logging----------------------------------
+wandb
+tensorboard
+
+# Extras----------------------------------
+pycocotools>=2.0.2
+setuptools==59.5.0