requirements.txt

catalyst==20.9
causal-conv1d==1.2.0
einops==0.7.0
fvcore==0.1.5.post20221221
imageio==2.34.0
importlib-metadata==7.1.0
lightning==2.2.2
lightning-utilities==0.11.2
mamba-ssm
matplotlib==3.7.1
matplotlib-inline==0.1.6
mmcv==2.1.0
mmdet==3.3.0
mmengine==0.10.1
mmpretrain==1.2.0
mmsegmentation==1.2.2
ninja==1.11.1.1
numpy==1.24.4
opencv-python==4.9.0.80
opencv-python-headless==4.9.0.80
pandas==2.0.3
pandocfilters==1.5.0
parso==0.8.3
Pillow==9.4.0
pytorch-lightning==2.2.2
scikit-image==0.21.0
scikit-learn==1.3.2
scipy==1.10.1
seaborn==0.13.2
selective-scan @ file:///UNetMamba/unetmamba_model/kernels/selective_scan
timm==0.4.12
tokenizers==0.19.1
torch @ http://download.pytorch.org/whl/cu118/torch-2.0.0%2Bcu118-cp38-cp38-linux_x86_64.whl
torchaudio==2.0.1
torchmetrics==1.3.2
torchvision @ http://download.pytorch.org/whl/cu118/torchvision-0.15.1%2Bcu118-cp38-cp38-linux_x86_64.whl
tqdm==4.57.0
transformers==4.40.1
triton==2.0.0