ABBSPO/requirements.txt at main · KAIST-VICLab/ABBSPO · GitHub

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
# ============================
# Core deep learning stack
# ============================
torch==1.13.1
torchvision==0.14.1
torchaudio==0.13.1

mmengine==0.10.0
mmdet==3.0.0

# ============================
# Numerical / image libraries
# ============================
numpy==1.24.3
scipy==1.10.1
pandas==2.0.3

opencv-python==4.10.0.84
Pillow==10.4.0
scikit-image==0.21.0
scikit-learn==1.3.2
matplotlib==3.7.5
pycocotools==2.0.7
shapely==2.0.6

# ============================
# Utils / training helpers
# ============================
tqdm==4.65.2
pyyaml==6.0.2
tabulate==0.9.0
termcolor==2.4.0
terminaltables==3.1.10
rich==13.4.2
addict==2.4.0

# ============================
# Loss, metrics, FLOPs, etc.
# ============================
piq==0.8.0
ptflops==0.7.3
imageio==2.35.1
imagecorruptions==1.1.2

# ============================
# Dev / formatting (optional)
# ============================
yapf==0.40.1

# NOTE:
# mmcv must be installed separately with CUDA support.
# See README (Environment Setup).