Views
No views yet
1conda create --name rscd python=3.8
2conda activate rscd
3conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.8 -c pytorch -c nvidia
4pip install pytorch-lightning==2.0.5
5pip install scikit-image==0.19.3 numpy==1.24.4
6pip install torchmetrics==1.0.1
7pip install -U catalyst==20.09
8pip install albumentations==1.3.1
9pip install einops==0.6.1
10pip install timm==0.6.7
11pip install addict==2.4.0
12pip install soundfile==0.12.1
13pip install ttach==0.0.3
14pip install prettytable==3.8.0
15pip install -U openmim
16pip install triton==2.0.0
17mim install mmcv
18pip install -U fvcore
19cd rscd/models/backbones/lib_mamba/kernels/selective_scan && pip install .1Please organize the datasets as follows:
2 rschangedetection
3 ├── rscd (code)
4 ├── work_dirs (save the model weights and training logs)
5 │ └─CLCD_BS4_epoch200 (dataset)
6 │ └─stnet (model)
7 │ └─version_0 (version)
8 │ │ └─ckpts
9 │ │ ├─test (the best ckpts in test set)
10 │ │ └─val (the best ckpts in validation set)
11 │ ├─log (tensorboard logs)
12 │ ├─train_metrics.txt (train & val results per epoch)
13 │ ├─test_metrics_max.txt (the best test results)
14 │ └─test_metrics_rest.txt (other test results)
15 └── data
16 ├── LEVIR_CD
17 │ ├── train
18 │ │ ├── A
19 │ │ │ └── images1.png
20 │ │ ├── B
21 │ │ │ └── images2.png
22 │ │ └── label
23 │ │ └── label.png
24 │ ├── val (the same with train)
25 │ └── test(the same with train)
26 ├── WHU_CD
27 │ ├── train
28 │ │ ├── image1
29 │ │ │ └── images1.png
30 │ │ ├── image2
31 │ │ │ └── images2.png
32 │ │ └── label
33 │ │ └── label.png
34 │ ├── val (the same with train)
35 │ └── test(the same with train)
36 └── CLCD (the same with WHU_CD)python train.py -c configs/mamba_cttf.py1python test.py \
2-c configs/mamba_cttf.py \
3--ckpt work_dirs/CLCD_BS4_epoch200/mamba_cttf/version_0/ckpts/test/epoch=156.ckpt \
4--output_dir work_dirs/CLCD_BS4_epoch200/mamba_cttf/version_0/ckpts/test \python tools/params_flops.py --size 256