forked from nmichlo/disent
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrequirements.txt
61 lines (48 loc) · 1.08 KB
/
requirements.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
pip>=21.0
# DATA SCIENCE & ML
# =================
numpy==1.20.2
pandas==1.2.4
# opencv-python==4.5.1.48
torch==1.8.1
torchvision==0.9.1
torch-optimizer==0.1.0
pytorch-lightning==1.2.7
# torchsort==0.1.4 # optional install
# pytorch-lightning-bolts==0.3.0
# tensorflow==2.4.1
# tensorflow-gpu==2.4.1
# tensorboard==2.4.1
scipy==1.6.2
scikit-learn==0.24.1
# xgboost==1.3.3
# lightgbm==3.1.1
# IMAGE AUGMENTATION
# ==================
kornia==0.5.0 # $ pip install git+https://github.com/kornia/kornia
# imgaug==0.4.0
# albumentations==0.5.2
# INPUT / OUTPUT
# ================
# psutil==5.8.0
# pillow==8.1.0
imageio==2.9.0 # required for wandb video logging
moviepy==1.0.3 # required for wandb video logging
h5py>=2.10.0 # as of tensorflow 2.4 it does not support h5py 3+
# GRAPHING & DISPLAY
# ==================
matplotlib==3.4.1
plotly==4.14.3
# seaborn==0.11.1
tqdm==4.60.0
# streamlit==0.75.0
wandb==0.10.25
# UTILITY
# =======
# pytest==6.2.1
hydra-core==1.0.6
hydra-colorlog==1.0.1
# hydra-joblib-launcher==1.1.1
hydra-submitit-launcher==1.1.1
# submitit==1.1.5
Deprecated==1.2.12