circleLZY commited on
Commit
b1b30c0
·
verified ·
1 Parent(s): 2b79cff

Upload 50 files

Browse files
Files changed (50) hide show
  1. HANet/distill/best_mIoU_iter_90000.pth +3 -0
  2. HANet/distill/config.py +51 -0
  3. HANet/initial/best_mIoU_iter_195000.pth +3 -0
  4. HANet/initial/config.py +3 -0
  5. HANet/large/best_mIoU_iter_122000.pth +3 -0
  6. HANet/large/config.py +29 -0
  7. HANet/medium/best_mIoU_iter_74000.pth +3 -0
  8. HANet/medium/config.py +29 -0
  9. HANet/small/best_mIoU_iter_158000.pth +3 -0
  10. HANet/small/config.py +29 -0
  11. IFN/distill/best_mIoU_iter_84000.pth +3 -0
  12. IFN/distill/config.py +51 -0
  13. IFN/initial/best_mIoU_iter_110000.pth +3 -0
  14. IFN/initial/config.py +3 -0
  15. IFN/large/best_mIoU_iter_173000.pth +3 -0
  16. IFN/large/config.py +29 -0
  17. IFN/medium/best_mIoU_iter_7000.pth +3 -0
  18. IFN/medium/config.py +29 -0
  19. IFN/small/best_mIoU_iter_145000.pth +3 -0
  20. IFN/small/config.py +29 -0
  21. LightCDNet/distill/best_mIoU_iter_168000.pth +3 -0
  22. LightCDNet/distill/config.py +65 -0
  23. LightCDNet/initial/best_mIoU_iter_195000.pth +3 -0
  24. LightCDNet/initial/config.py +19 -0
  25. LightCDNet/large/best_mIoU_iter_185000.pth +3 -0
  26. LightCDNet/large/config.py +50 -0
  27. LightCDNet/medium/best_mIoU_iter_199000.pth +3 -0
  28. LightCDNet/medium/config.py +45 -0
  29. LightCDNet/small/best_mIoU_iter_193000.pth +3 -0
  30. LightCDNet/small/config.py +45 -0
  31. SNUNet/distill/best_mIoU_iter_88000.pth +3 -0
  32. SNUNet/distill/config.py +51 -0
  33. SNUNet/initial/best_mIoU_iter_160000.pth +3 -0
  34. SNUNet/initial/config.py +9 -0
  35. SNUNet/large/best_mIoU_iter_62000.pth +3 -0
  36. SNUNet/large/config.py +34 -0
  37. SNUNet/medium/best_mIoU_iter_20000.pth +3 -0
  38. SNUNet/medium/config.py +34 -0
  39. SNUNet/small/best_mIoU_iter_177000.pth +3 -0
  40. SNUNet/small/config.py +34 -0
  41. STANet/distill/best_mIoU_iter_46000.pth +3 -0
  42. STANet/distill/config.py +51 -0
  43. STANet/initial/best_mIoU_iter_130000.pth +3 -0
  44. STANet/initial/config.py +9 -0
  45. STANet/large/best_mIoU_iter_24000.pth +3 -0
  46. STANet/large/config.py +36 -0
  47. STANet/medium/best_mIoU_iter_28000.pth +3 -0
  48. STANet/medium/config.py +36 -0
  49. STANet/small/best_mIoU_iter_172000.pth +3 -0
  50. STANet/small/config.py +36 -0
HANet/distill/best_mIoU_iter_90000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5dfac04383de944b87b230e9796ed2622ecc057d30323ed8e42d5609bf4cc2bc
3
+ size 60601169
HANet/distill/config.py ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _base_ = [
2
+ '/home/liuziyuan/proj/rmcd-kd/configs/_base_/models/KD-hanet.py',
3
+ '/home/liuziyuan/proj/rmcd-kd/configs/common/standard_512x512_200k_cgwx.py']
4
+
5
+ dataset_type = 'LEVIR_CD_Dataset'
6
+ data_root = '/nas/datasets/lzy/RS-ChangeDetection/CGWX'
7
+
8
+ crop_size = (512, 512)
9
+
10
+ checkpoint_student = '/nas/datasets/lzy/RS-ChangeDetection/checkpoints_distill/HANet/teacher_ckpt/initial/best_mIoU_iter_195000.pth'
11
+ checkpoint_teacher_l = '/nas/datasets/lzy/RS-ChangeDetection/checkpoints_distill/HANet/teacher_ckpt/large/best_mIoU_iter_122000.pth'
12
+ checkpoint_teacher_m = '/nas/datasets/lzy/RS-ChangeDetection/checkpoints_distill/HANet/teacher_ckpt/medium/best_mIoU_iter_74000.pth'
13
+ checkpoint_teacher_s = '/nas/datasets/lzy/RS-ChangeDetection/checkpoints_distill/HANet/teacher_ckpt/small/best_mIoU_iter_158000.pth'
14
+
15
+ model = dict(
16
+ # student
17
+ init_cfg=dict(type='Pretrained', checkpoint=checkpoint_student),
18
+ # teacher large
19
+ init_cfg_t_l = dict(type='Pretrained', checkpoint=checkpoint_teacher_l),
20
+ # teacher medium
21
+ init_cfg_t_m = dict(type='Pretrained', checkpoint=checkpoint_teacher_m),
22
+ # teacher small
23
+ init_cfg_t_s = dict(type='Pretrained', checkpoint=checkpoint_teacher_s),
24
+ )
25
+
26
+ param_scheduler = [
27
+ dict(
28
+ type='LinearLR', start_factor=1e-6, by_epoch=False, begin=0, end=1000),
29
+ dict(
30
+ type='PolyLR',
31
+ power=1.0,
32
+ begin=1000,
33
+ end=100000,
34
+ eta_min=0.0,
35
+ by_epoch=False,
36
+ )
37
+ ]
38
+
39
+ # training schedule for 100k
40
+ train_cfg = dict(type='IterBasedTrainLoop', max_iters=100000, val_interval=1000)
41
+ val_cfg = dict(type='ValLoop')
42
+ test_cfg = dict(type='TestLoop')
43
+ default_hooks = dict(
44
+ timer=dict(type='IterTimerHook'),
45
+ logger=dict(type='LoggerHook', interval=100, log_metric_by_epoch=False),
46
+ param_scheduler=dict(type='ParamSchedulerHook'),
47
+ checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=1000,
48
+ save_best='mIoU'),
49
+ sampler_seed=dict(type='DistSamplerSeedHook'),
50
+ visualization=dict(type='CDVisualizationHook', interval=1,
51
+ img_shape=(512, 512, 3)))
HANet/initial/best_mIoU_iter_195000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ae0deb75f8bb6140996ceae6a7fd11525781ee9440d32f24e15f234a78dc04e9
3
+ size 33597877
HANet/initial/config.py ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ _base_ = [
2
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/_base_/models/hanet.py',
3
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/common/standard_512x512_200k_cgwx.py']
HANet/large/best_mIoU_iter_122000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c042459354a0ce90fe0690d44dc360e645b80a78c6cec9ca69f338234812b475
3
+ size 25744885
HANet/large/config.py ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _base_ = [
2
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/_base_/models/hanet.py',
3
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/common/train_large_512x512_100k_cgwx.py']
4
+
5
+ param_scheduler = [
6
+ dict(
7
+ type='LinearLR', start_factor=1e-6, by_epoch=False, begin=0, end=1000),
8
+ dict(
9
+ type='PolyLR',
10
+ power=1.0,
11
+ begin=1000,
12
+ end=200000,
13
+ eta_min=0.0,
14
+ by_epoch=False,
15
+ )
16
+ ]
17
+ # training schedule for 100k
18
+ train_cfg = dict(type='IterBasedTrainLoop', max_iters=200000, val_interval=1000)
19
+ val_cfg = dict(type='ValLoop')
20
+ test_cfg = dict(type='TestLoop')
21
+ default_hooks = dict(
22
+ timer=dict(type='IterTimerHook'),
23
+ logger=dict(type='LoggerHook', interval=100, log_metric_by_epoch=False),
24
+ param_scheduler=dict(type='ParamSchedulerHook'),
25
+ checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=1000,
26
+ save_best='mIoU'),
27
+ sampler_seed=dict(type='DistSamplerSeedHook'),
28
+ visualization=dict(type='CDVisualizationHook', interval=1,
29
+ img_shape=(512, 512, 3)))
HANet/medium/best_mIoU_iter_74000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:30ff8cced9397dbe9f8644bf98fa413cf99e3456b9f49f88deabe4c7a4c8b2d7
3
+ size 20387381
HANet/medium/config.py ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _base_ = [
2
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/_base_/models/hanet.py',
3
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/common/train_medium_512x512_100k_cgwx.py']
4
+
5
+ param_scheduler = [
6
+ dict(
7
+ type='LinearLR', start_factor=1e-6, by_epoch=False, begin=0, end=1000),
8
+ dict(
9
+ type='PolyLR',
10
+ power=1.0,
11
+ begin=1000,
12
+ end=200000,
13
+ eta_min=0.0,
14
+ by_epoch=False,
15
+ )
16
+ ]
17
+ # training schedule for 100k
18
+ train_cfg = dict(type='IterBasedTrainLoop', max_iters=200000, val_interval=1000)
19
+ val_cfg = dict(type='ValLoop')
20
+ test_cfg = dict(type='TestLoop')
21
+ default_hooks = dict(
22
+ timer=dict(type='IterTimerHook'),
23
+ logger=dict(type='LoggerHook', interval=100, log_metric_by_epoch=False),
24
+ param_scheduler=dict(type='ParamSchedulerHook'),
25
+ checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=1000,
26
+ save_best='mIoU'),
27
+ sampler_seed=dict(type='DistSamplerSeedHook'),
28
+ visualization=dict(type='CDVisualizationHook', interval=1,
29
+ img_shape=(512, 512, 3)))
HANet/small/best_mIoU_iter_158000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bae9176f27257c6efed0f01bb03ea3faa3e4bea4f46ceba355fad2d99b3d9d0b
3
+ size 29915893
HANet/small/config.py ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _base_ = [
2
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/_base_/models/hanet.py',
3
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/common/train_small_512x512_100k_cgwx.py']
4
+
5
+ param_scheduler = [
6
+ dict(
7
+ type='LinearLR', start_factor=1e-6, by_epoch=False, begin=0, end=1000),
8
+ dict(
9
+ type='PolyLR',
10
+ power=1.0,
11
+ begin=1000,
12
+ end=200000,
13
+ eta_min=0.0,
14
+ by_epoch=False,
15
+ )
16
+ ]
17
+ # training schedule for 100k
18
+ train_cfg = dict(type='IterBasedTrainLoop', max_iters=200000, val_interval=1000)
19
+ val_cfg = dict(type='ValLoop')
20
+ test_cfg = dict(type='TestLoop')
21
+ default_hooks = dict(
22
+ timer=dict(type='IterTimerHook'),
23
+ logger=dict(type='LoggerHook', interval=100, log_metric_by_epoch=False),
24
+ param_scheduler=dict(type='ParamSchedulerHook'),
25
+ checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=1000,
26
+ save_best='mIoU'),
27
+ sampler_seed=dict(type='DistSamplerSeedHook'),
28
+ visualization=dict(type='CDVisualizationHook', interval=1,
29
+ img_shape=(512, 512, 3)))
IFN/distill/best_mIoU_iter_84000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b905b44435c234f5ce7eac375de9cc07ea1fc814d9dd7e1b4e600e39eaef72d7
3
+ size 830631809
IFN/distill/config.py ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _base_ = [
2
+ '/home/liuziyuan/proj/rmcd-kd/configs/_base_/models/KD-ifn.py',
3
+ '/home/liuziyuan/proj/rmcd-kd/configs/common/standard_512x512_200k_cgwx.py']
4
+
5
+ dataset_type = 'LEVIR_CD_Dataset'
6
+ data_root = '/nas/datasets/lzy/RS-ChangeDetection/CGWX'
7
+
8
+ crop_size = (512, 512)
9
+
10
+ checkpoint_student = '/nas/datasets/lzy/RS-ChangeDetection/checkpoints_distill/IFN/teacher_ckpt/initial/best_mIoU_iter_110000.pth'
11
+ checkpoint_teacher_l = '/nas/datasets/lzy/RS-ChangeDetection/checkpoints_distill/IFN/teacher_ckpt/large/best_mIoU_iter_173000.pth'
12
+ checkpoint_teacher_m = '/nas/datasets/lzy/RS-ChangeDetection/checkpoints_distill/IFN/teacher_ckpt/medium/best_mIoU_iter_7000.pth'
13
+ checkpoint_teacher_s = '/nas/datasets/lzy/RS-ChangeDetection/checkpoints_distill/IFN/teacher_ckpt/small/best_mIoU_iter_145000.pth'
14
+
15
+ model = dict(
16
+ # student
17
+ init_cfg=dict(type='Pretrained', checkpoint=checkpoint_student),
18
+ # teacher large
19
+ init_cfg_t_l = dict(type='Pretrained', checkpoint=checkpoint_teacher_l),
20
+ # teacher medium
21
+ init_cfg_t_m = dict(type='Pretrained', checkpoint=checkpoint_teacher_m),
22
+ # teacher small
23
+ init_cfg_t_s = dict(type='Pretrained', checkpoint=checkpoint_teacher_s)
24
+ )
25
+
26
+ param_scheduler = [
27
+ dict(
28
+ type='LinearLR', start_factor=1e-6, by_epoch=False, begin=0, end=1000),
29
+ dict(
30
+ type='PolyLR',
31
+ power=1.0,
32
+ begin=1000,
33
+ end=100000,
34
+ eta_min=0.0,
35
+ by_epoch=False,
36
+ )
37
+ ]
38
+
39
+ # training schedule for 100k
40
+ train_cfg = dict(type='IterBasedTrainLoop', max_iters=100000, val_interval=1000)
41
+ val_cfg = dict(type='ValLoop')
42
+ test_cfg = dict(type='TestLoop')
43
+ default_hooks = dict(
44
+ timer=dict(type='IterTimerHook'),
45
+ logger=dict(type='LoggerHook', interval=50, log_metric_by_epoch=False),
46
+ param_scheduler=dict(type='ParamSchedulerHook'),
47
+ checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=1000,
48
+ save_best='mIoU'),
49
+ sampler_seed=dict(type='DistSamplerSeedHook'),
50
+ visualization=dict(type='CDVisualizationHook', interval=1,
51
+ img_shape=(512, 512, 3)))
IFN/initial/best_mIoU_iter_110000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c08a12b4ec775867d61cf4971c3f7a91838eefcad4376cace42fb5820b7ab37a
3
+ size 225192561
IFN/initial/config.py ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ _base_ = [
2
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/_base_/models/ifn.py',
3
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/common/standard_512x512_200k_cgwx.py']
IFN/large/best_mIoU_iter_173000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f5c639fd180ad1d490523bf676e6bd88637d95a6d422dfc9bfbe88bc64cca103
3
+ size 237871089
IFN/large/config.py ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _base_ = [
2
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/_base_/models/ifn.py',
3
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/common/train_large_512x512_100k_cgwx.py']
4
+
5
+ param_scheduler = [
6
+ dict(
7
+ type='LinearLR', start_factor=1e-6, by_epoch=False, begin=0, end=1000),
8
+ dict(
9
+ type='PolyLR',
10
+ power=1.0,
11
+ begin=1000,
12
+ end=200000,
13
+ eta_min=0.0,
14
+ by_epoch=False,
15
+ )
16
+ ]
17
+ # training schedule for 100k
18
+ train_cfg = dict(type='IterBasedTrainLoop', max_iters=200000, val_interval=1000)
19
+ val_cfg = dict(type='ValLoop')
20
+ test_cfg = dict(type='TestLoop')
21
+ default_hooks = dict(
22
+ timer=dict(type='IterTimerHook'),
23
+ logger=dict(type='LoggerHook', interval=100, log_metric_by_epoch=False),
24
+ param_scheduler=dict(type='ParamSchedulerHook'),
25
+ checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=1000,
26
+ save_best='mIoU'),
27
+ sampler_seed=dict(type='DistSamplerSeedHook'),
28
+ visualization=dict(type='CDVisualizationHook', interval=1,
29
+ img_shape=(512, 512, 3)))
IFN/medium/best_mIoU_iter_7000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3cf95241699981f4100a3bc6cca71dfe67bc976c96373647e70657d3b1e6f42d
3
+ size 204405809
IFN/medium/config.py ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _base_ = [
2
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/_base_/models/ifn.py',
3
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/common/train_medium_512x512_100k_cgwx.py']
4
+
5
+ param_scheduler = [
6
+ dict(
7
+ type='LinearLR', start_factor=1e-6, by_epoch=False, begin=0, end=1000),
8
+ dict(
9
+ type='PolyLR',
10
+ power=1.0,
11
+ begin=1000,
12
+ end=200000,
13
+ eta_min=0.0,
14
+ by_epoch=False,
15
+ )
16
+ ]
17
+ # training schedule for 100k
18
+ train_cfg = dict(type='IterBasedTrainLoop', max_iters=200000, val_interval=1000)
19
+ val_cfg = dict(type='ValLoop')
20
+ test_cfg = dict(type='TestLoop')
21
+ default_hooks = dict(
22
+ timer=dict(type='IterTimerHook'),
23
+ logger=dict(type='LoggerHook', interval=100, log_metric_by_epoch=False),
24
+ param_scheduler=dict(type='ParamSchedulerHook'),
25
+ checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=1000,
26
+ save_best='mIoU'),
27
+ sampler_seed=dict(type='DistSamplerSeedHook'),
28
+ visualization=dict(type='CDVisualizationHook', interval=1,
29
+ img_shape=(512, 512, 3)))
IFN/small/best_mIoU_iter_145000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b700377d58bc681cb5855024ed9cc310bdd04eabb433bf017c39b597601a044e
3
+ size 232378097
IFN/small/config.py ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _base_ = [
2
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/_base_/models/ifn.py',
3
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/common/train_small_512x512_100k_cgwx.py']
4
+
5
+ param_scheduler = [
6
+ dict(
7
+ type='LinearLR', start_factor=1e-6, by_epoch=False, begin=0, end=1000),
8
+ dict(
9
+ type='PolyLR',
10
+ power=1.0,
11
+ begin=1000,
12
+ end=200000,
13
+ eta_min=0.0,
14
+ by_epoch=False,
15
+ )
16
+ ]
17
+ # training schedule for 100k
18
+ train_cfg = dict(type='IterBasedTrainLoop', max_iters=200000, val_interval=1000)
19
+ val_cfg = dict(type='ValLoop')
20
+ test_cfg = dict(type='TestLoop')
21
+ default_hooks = dict(
22
+ timer=dict(type='IterTimerHook'),
23
+ logger=dict(type='LoggerHook', interval=100, log_metric_by_epoch=False),
24
+ param_scheduler=dict(type='ParamSchedulerHook'),
25
+ checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=1000,
26
+ save_best='mIoU'),
27
+ sampler_seed=dict(type='DistSamplerSeedHook'),
28
+ visualization=dict(type='CDVisualizationHook', interval=1,
29
+ img_shape=(512, 512, 3)))
LightCDNet/distill/best_mIoU_iter_168000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3305c6b9725ca736020901db7de725b4b4789783c4179431c293c730ce31f4ea
3
+ size 36647731
LightCDNet/distill/config.py ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _base_ = [
2
+ '/home/liuziyuan/proj/rmcd-kd/configs/_base_/models/KD-lightcdnet.py',
3
+ '/home/liuziyuan/proj/rmcd-kd/configs/common/standard_512x512_200k_cgwx.py']
4
+
5
+ dataset_type = 'LEVIR_CD_Dataset'
6
+ data_root = '/nas/datasets/lzy/RS-ChangeDetection/CGWX'
7
+
8
+ crop_size = (512, 512)
9
+
10
+ checkpoint_student = '/nas/datasets/lzy/RS-ChangeDetection/checkpoints_distill/LightCDNet/teacher_ckpt/initial/best_mIoU_iter_195000.pth'
11
+ checkpoint_teacher_l = '/nas/datasets/lzy/RS-ChangeDetection/checkpoints_distill/LightCDNet/teacher_ckpt/large/best_mIoU_iter_185000.pth'
12
+ checkpoint_teacher_m = '/nas/datasets/lzy/RS-ChangeDetection/checkpoints_distill/LightCDNet/teacher_ckpt/medium/best_mIoU_iter_199000.pth'
13
+ checkpoint_teacher_s = '/nas/datasets/lzy/RS-ChangeDetection/checkpoints_distill/LightCDNet/teacher_ckpt/small/best_mIoU_iter_193000.pth'
14
+
15
+ model = dict(
16
+ # student
17
+ init_cfg=dict(type='Pretrained', checkpoint=checkpoint_student),
18
+ # teacher large
19
+ init_cfg_t_l = dict(type='Pretrained', checkpoint=checkpoint_teacher_l),
20
+ # teacher medium
21
+ init_cfg_t_m = dict(type='Pretrained', checkpoint=checkpoint_teacher_m),
22
+ # teacher small
23
+ init_cfg_t_s = dict(type='Pretrained', checkpoint=checkpoint_teacher_s),
24
+
25
+ decode_head=dict(
26
+ sampler=dict(type='mmseg.OHEMPixelSampler', thresh=0.7, min_kept=100000)))
27
+
28
+ # optimizer
29
+ optimizer = dict(
30
+ type='AdamW',
31
+ lr=0.003,
32
+ betas=(0.9, 0.999),
33
+ weight_decay=0.05)
34
+
35
+ optim_wrapper = dict(
36
+ _delete_=True,
37
+ type='OptimWrapper',
38
+ optimizer=optimizer)
39
+
40
+ param_scheduler = [
41
+ dict(
42
+ type='LinearLR', start_factor=1e-6, by_epoch=False, begin=0, end=1000),
43
+ dict(
44
+ type='PolyLR',
45
+ power=1.0,
46
+ begin=1000,
47
+ end=100000,
48
+ eta_min=0.0,
49
+ by_epoch=False,
50
+ )
51
+ ]
52
+
53
+ # training schedule for 100k
54
+ train_cfg = dict(type='IterBasedTrainLoop', max_iters=100000, val_interval=1000)
55
+ val_cfg = dict(type='ValLoop')
56
+ test_cfg = dict(type='TestLoop')
57
+ default_hooks = dict(
58
+ timer=dict(type='IterTimerHook'),
59
+ logger=dict(type='LoggerHook', interval=100, log_metric_by_epoch=False),
60
+ param_scheduler=dict(type='ParamSchedulerHook'),
61
+ checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=1000,
62
+ save_best='mIoU'),
63
+ sampler_seed=dict(type='DistSamplerSeedHook'),
64
+ visualization=dict(type='CDVisualizationHook', interval=1,
65
+ img_shape=(512, 512, 3)))
LightCDNet/initial/best_mIoU_iter_195000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6470f753ab113e6f8e92eb11a57966d265d90b29eee60ce0e90250b58364a6c9
3
+ size 30051619
LightCDNet/initial/config.py ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _base_ = [
2
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/_base_/models/lightcdnet.py',
3
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/common/standard_512x512_200k_cgwx.py']
4
+
5
+ model = dict(
6
+ decode_head=dict(
7
+ sampler=dict(type='mmseg.OHEMPixelSampler', thresh=0.7, min_kept=100000)))
8
+
9
+ # optimizer
10
+ optimizer = dict(
11
+ type='AdamW',
12
+ lr=0.003,
13
+ betas=(0.9, 0.999),
14
+ weight_decay=0.05)
15
+
16
+ optim_wrapper = dict(
17
+ _delete_=True,
18
+ type='OptimWrapper',
19
+ optimizer=optimizer)
LightCDNet/large/best_mIoU_iter_185000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:70d1ec7952db963d61328bdd4c03da628b4ec4a4721baed71f35712fd6e5d2e9
3
+ size 28566435
LightCDNet/large/config.py ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _base_ = [
2
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/_base_/models/lightcdnet.py',
3
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/common/train_large_512x512_100k_cgwx.py']
4
+
5
+ dataset_type = 'LEVIR_CD_Dataset'
6
+ data_root = '/nas/datasets/lzy/RS-ChangeDetection/CGWX'
7
+
8
+ crop_size = (512, 512)
9
+
10
+ model = dict(
11
+ decode_head=dict(
12
+ sampler=dict(type='mmseg.OHEMPixelSampler', thresh=0.7, min_kept=100000)))
13
+
14
+ # optimizer
15
+ optimizer = dict(
16
+ type='AdamW',
17
+ lr=0.003,
18
+ betas=(0.9, 0.999),
19
+ weight_decay=0.05)
20
+
21
+ optim_wrapper = dict(
22
+ _delete_=True,
23
+ type='OptimWrapper',
24
+ optimizer=optimizer)
25
+
26
+ param_scheduler = [
27
+ dict(
28
+ type='LinearLR', start_factor=1e-6, by_epoch=False, begin=0, end=1000),
29
+ dict(
30
+ type='PolyLR',
31
+ power=1.0,
32
+ begin=1000,
33
+ end=200000,
34
+ eta_min=0.0,
35
+ by_epoch=False,
36
+ )
37
+ ]
38
+ # training schedule for 100k
39
+ train_cfg = dict(type='IterBasedTrainLoop', max_iters=200000, val_interval=1000)
40
+ val_cfg = dict(type='ValLoop')
41
+ test_cfg = dict(type='TestLoop')
42
+ default_hooks = dict(
43
+ timer=dict(type='IterTimerHook'),
44
+ logger=dict(type='LoggerHook', interval=50, log_metric_by_epoch=False),
45
+ param_scheduler=dict(type='ParamSchedulerHook'),
46
+ checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=1000,
47
+ save_best='mIoU'),
48
+ sampler_seed=dict(type='DistSamplerSeedHook'),
49
+ visualization=dict(type='CDVisualizationHook', interval=1,
50
+ img_shape=(512, 512, 3)))
LightCDNet/medium/best_mIoU_iter_199000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9cd26bb4fb37f6083f0b5643db5295c82bfea948e45aabcc54b30181b5909e47
3
+ size 30642147
LightCDNet/medium/config.py ADDED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _base_ = [
2
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/_base_/models/lightcdnet.py',
3
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/common/train_medium_512x512_100k_cgwx.py']
4
+
5
+ model = dict(
6
+ decode_head=dict(
7
+ sampler=dict(type='mmseg.OHEMPixelSampler', thresh=0.7, min_kept=100000)))
8
+
9
+ # optimizer
10
+ optimizer = dict(
11
+ type='AdamW',
12
+ lr=0.003,
13
+ betas=(0.9, 0.999),
14
+ weight_decay=0.05)
15
+
16
+ optim_wrapper = dict(
17
+ _delete_=True,
18
+ type='OptimWrapper',
19
+ optimizer=optimizer)
20
+
21
+ param_scheduler = [
22
+ dict(
23
+ type='LinearLR', start_factor=1e-6, by_epoch=False, begin=0, end=1000),
24
+ dict(
25
+ type='PolyLR',
26
+ power=1.0,
27
+ begin=1000,
28
+ end=200000,
29
+ eta_min=0.0,
30
+ by_epoch=False,
31
+ )
32
+ ]
33
+ # training schedule for 100k
34
+ train_cfg = dict(type='IterBasedTrainLoop', max_iters=200000, val_interval=1000)
35
+ val_cfg = dict(type='ValLoop')
36
+ test_cfg = dict(type='TestLoop')
37
+ default_hooks = dict(
38
+ timer=dict(type='IterTimerHook'),
39
+ logger=dict(type='LoggerHook', interval=50, log_metric_by_epoch=False),
40
+ param_scheduler=dict(type='ParamSchedulerHook'),
41
+ checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=1000,
42
+ save_best='mIoU'),
43
+ sampler_seed=dict(type='DistSamplerSeedHook'),
44
+ visualization=dict(type='CDVisualizationHook', interval=1,
45
+ img_shape=(512, 512, 3)))
LightCDNet/small/best_mIoU_iter_193000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d593f4c65819f287d4d5de18a1786470a418f1129ad034989239e879f931346c
3
+ size 30034275
LightCDNet/small/config.py ADDED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _base_ = [
2
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/_base_/models/lightcdnet.py',
3
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/common/train_small_512x512_100k_cgwx.py']
4
+
5
+ model = dict(
6
+ decode_head=dict(
7
+ sampler=dict(type='mmseg.OHEMPixelSampler', thresh=0.7, min_kept=100000)))
8
+
9
+ # optimizer
10
+ optimizer = dict(
11
+ type='AdamW',
12
+ lr=0.003,
13
+ betas=(0.9, 0.999),
14
+ weight_decay=0.05)
15
+
16
+ optim_wrapper = dict(
17
+ _delete_=True,
18
+ type='OptimWrapper',
19
+ optimizer=optimizer)
20
+
21
+ param_scheduler = [
22
+ dict(
23
+ type='LinearLR', start_factor=1e-6, by_epoch=False, begin=0, end=1000),
24
+ dict(
25
+ type='PolyLR',
26
+ power=1.0,
27
+ begin=1000,
28
+ end=200000,
29
+ eta_min=0.0,
30
+ by_epoch=False,
31
+ )
32
+ ]
33
+ # training schedule for 100k
34
+ train_cfg = dict(type='IterBasedTrainLoop', max_iters=200000, val_interval=1000)
35
+ val_cfg = dict(type='ValLoop')
36
+ test_cfg = dict(type='TestLoop')
37
+ default_hooks = dict(
38
+ timer=dict(type='IterTimerHook'),
39
+ logger=dict(type='LoggerHook', interval=50, log_metric_by_epoch=False),
40
+ param_scheduler=dict(type='ParamSchedulerHook'),
41
+ checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=1000,
42
+ save_best='mIoU'),
43
+ sampler_seed=dict(type='DistSamplerSeedHook'),
44
+ visualization=dict(type='CDVisualizationHook', interval=1,
45
+ img_shape=(512, 512, 3)))
SNUNet/distill/best_mIoU_iter_88000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1b4f63d2f042225e9c0bc5e4a0916f5f40c0c0536802b5dca35f3d1fd917863a
3
+ size 60162785
SNUNet/distill/config.py ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _base_ = [
2
+ '/home/liuziyuan/proj/rmcd-kd/configs/_base_/models/KD-snunet_c16.py',
3
+ '/home/liuziyuan/proj/rmcd-kd/configs/common/standard_512x512_200k_cgwx.py']
4
+
5
+ dataset_type = 'LEVIR_CD_Dataset'
6
+ data_root = '/nas/datasets/lzy/RS-ChangeDetection/CGWX'
7
+
8
+ crop_size = (512, 512)
9
+
10
+ checkpoint_student = '/nas/datasets/lzy/RS-ChangeDetection/checkpoints_distill/SNUNet/teacher_ckpt/initial/best_mIoU_iter_160000.pth'
11
+ checkpoint_teacher_l = '/nas/datasets/lzy/RS-ChangeDetection/checkpoints_distill/SNUNet/teacher_ckpt/large/best_mIoU_iter_62000.pth'
12
+ checkpoint_teacher_m = '/nas/datasets/lzy/RS-ChangeDetection/checkpoints_distill/SNUNet/teacher_ckpt/medium/best_mIoU_iter_20000.pth'
13
+ checkpoint_teacher_s = '/nas/datasets/lzy/RS-ChangeDetection/checkpoints_distill/SNUNet/teacher_ckpt/small/best_mIoU_iter_177000.pth'
14
+
15
+ model = dict(
16
+ # student
17
+ init_cfg=dict(type='Pretrained', checkpoint=checkpoint_student),
18
+ # teacher large
19
+ init_cfg_t_l = dict(type='Pretrained', checkpoint=checkpoint_teacher_l),
20
+ # teacher medium
21
+ init_cfg_t_m = dict(type='Pretrained', checkpoint=checkpoint_teacher_m),
22
+ # teacher small
23
+ init_cfg_t_s = dict(type='Pretrained', checkpoint=checkpoint_teacher_s),
24
+ )
25
+
26
+ param_scheduler = [
27
+ dict(
28
+ type='LinearLR', start_factor=1e-6, by_epoch=False, begin=0, end=1000),
29
+ dict(
30
+ type='PolyLR',
31
+ power=1.0,
32
+ begin=1000,
33
+ end=100000,
34
+ eta_min=0.0,
35
+ by_epoch=False,
36
+ )
37
+ ]
38
+
39
+ # training schedule for 100k
40
+ train_cfg = dict(type='IterBasedTrainLoop', max_iters=100000, val_interval=1000)
41
+ val_cfg = dict(type='ValLoop')
42
+ test_cfg = dict(type='TestLoop')
43
+ default_hooks = dict(
44
+ timer=dict(type='IterTimerHook'),
45
+ logger=dict(type='LoggerHook', interval=100, log_metric_by_epoch=False),
46
+ param_scheduler=dict(type='ParamSchedulerHook'),
47
+ checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=1000,
48
+ save_best='mIoU'),
49
+ sampler_seed=dict(type='DistSamplerSeedHook'),
50
+ visualization=dict(type='CDVisualizationHook', interval=1,
51
+ img_shape=(512, 512, 3)))
SNUNet/initial/best_mIoU_iter_160000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fa67db15e5c8464c23a9414b3346d3bfa30e01521f061dc39b74026f23167452
3
+ size 29803689
SNUNet/initial/config.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ _base_ = [
2
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/_base_/models/snunet_c16.py',
3
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/common/standard_512x512_200k_cgwx.py']
4
+
5
+
6
+ dataset_type = 'LEVIR_CD_Dataset'
7
+ data_root = '/nas/datasets/lzy/RS-ChangeDetection/CGWX'
8
+
9
+ crop_size = (512, 512)
SNUNet/large/best_mIoU_iter_62000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0de5306faf18bb1647f429425ee576b0fdc47b33e15d0f3e029700c522ee471e
3
+ size 19018217
SNUNet/large/config.py ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _base_ = [
2
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/_base_/models/snunet_c16.py',
3
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/common/train_large_512x512_100k_cgwx.py']
4
+
5
+ dataset_type = 'LEVIR_CD_Dataset'
6
+ data_root = '/nas/datasets/lzy/RS-ChangeDetection/CGWX'
7
+
8
+ crop_size = (512, 512)
9
+
10
+ param_scheduler = [
11
+ dict(
12
+ type='LinearLR', start_factor=1e-6, by_epoch=False, begin=0, end=1000),
13
+ dict(
14
+ type='PolyLR',
15
+ power=1.0,
16
+ begin=1000,
17
+ end=200000,
18
+ eta_min=0.0,
19
+ by_epoch=False,
20
+ )
21
+ ]
22
+
23
+ # training schedule for 100k
24
+ train_cfg = dict(type='IterBasedTrainLoop', max_iters=200000, val_interval=1000)
25
+
26
+ default_hooks = dict(
27
+ timer=dict(type='IterTimerHook'),
28
+ logger=dict(type='LoggerHook', interval=100, log_metric_by_epoch=False),
29
+ param_scheduler=dict(type='ParamSchedulerHook'),
30
+ checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=1000,
31
+ save_best='mIoU'),
32
+ sampler_seed=dict(type='DistSamplerSeedHook'),
33
+ visualization=dict(type='CDVisualizationHook', interval=1,
34
+ img_shape=(512, 512, 3)))
SNUNet/medium/best_mIoU_iter_20000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a5885437deb56fcb0d882c319422809bfd90306b3941f609c3a4d983fca40d4e
3
+ size 14375209
SNUNet/medium/config.py ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _base_ = [
2
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/_base_/models/snunet_c16.py',
3
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/common/train_medium_512x512_100k_cgwx.py']
4
+
5
+ dataset_type = 'LEVIR_CD_Dataset'
6
+ data_root = '/nas/datasets/lzy/RS-ChangeDetection/CGWX'
7
+
8
+ crop_size = (512, 512)
9
+
10
+ param_scheduler = [
11
+ dict(
12
+ type='LinearLR', start_factor=1e-6, by_epoch=False, begin=0, end=1000),
13
+ dict(
14
+ type='PolyLR',
15
+ power=1.0,
16
+ begin=1000,
17
+ end=200000,
18
+ eta_min=0.0,
19
+ by_epoch=False,
20
+ )
21
+ ]
22
+
23
+ # training schedule for 100k
24
+ train_cfg = dict(type='IterBasedTrainLoop', max_iters=200000, val_interval=1000)
25
+
26
+ default_hooks = dict(
27
+ timer=dict(type='IterTimerHook'),
28
+ logger=dict(type='LoggerHook', interval=100, log_metric_by_epoch=False),
29
+ param_scheduler=dict(type='ParamSchedulerHook'),
30
+ checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=1000,
31
+ save_best='mIoU'),
32
+ sampler_seed=dict(type='DistSamplerSeedHook'),
33
+ visualization=dict(type='CDVisualizationHook', interval=1,
34
+ img_shape=(512, 512, 3)))
SNUNet/small/best_mIoU_iter_177000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6d32a1f1a42d4a25947a469164e1dd6e340e1d7a128f01c2dd7fcb22e6479640
3
+ size 31962025
SNUNet/small/config.py ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _base_ = [
2
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/_base_/models/snunet_c16.py',
3
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/common/train_small_512x512_100k_cgwx.py']
4
+
5
+ dataset_type = 'LEVIR_CD_Dataset'
6
+ data_root = '/nas/datasets/lzy/RS-ChangeDetection/CGWX'
7
+
8
+ crop_size = (512, 512)
9
+
10
+ param_scheduler = [
11
+ dict(
12
+ type='LinearLR', start_factor=1e-6, by_epoch=False, begin=0, end=1000),
13
+ dict(
14
+ type='PolyLR',
15
+ power=1.0,
16
+ begin=1000,
17
+ end=200000,
18
+ eta_min=0.0,
19
+ by_epoch=False,
20
+ )
21
+ ]
22
+
23
+ # training schedule for 100k
24
+ train_cfg = dict(type='IterBasedTrainLoop', max_iters=200000, val_interval=1000)
25
+
26
+ default_hooks = dict(
27
+ timer=dict(type='IterTimerHook'),
28
+ logger=dict(type='LoggerHook', interval=100, log_metric_by_epoch=False),
29
+ param_scheduler=dict(type='ParamSchedulerHook'),
30
+ checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=1000,
31
+ save_best='mIoU'),
32
+ sampler_seed=dict(type='DistSamplerSeedHook'),
33
+ visualization=dict(type='CDVisualizationHook', interval=1,
34
+ img_shape=(512, 512, 3)))
STANet/distill/best_mIoU_iter_46000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:09117953d2d974b3a8aea59afd285967ddafde8099c0eb616cd941c89f9ae3cb
3
+ size 210657729
STANet/distill/config.py ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _base_ = [
2
+ '/home/liuziyuan/proj/rmcd-kd/configs/_base_/models/KD-stanet_r18.py',
3
+ '/home/liuziyuan/proj/rmcd-kd/configs/common/standard_512x512_200k_cgwx.py']
4
+
5
+ crop_size = (512, 512)
6
+
7
+ checkpoint_student = '/nas/datasets/lzy/RS-ChangeDetection/checkpoints_distill/STANet/teacher_ckpt/initial/best_mIoU_iter_130000.pth'
8
+ checkpoint_teacher_l = '/nas/datasets/lzy/RS-ChangeDetection/checkpoints_distill/STANet/teacher_ckpt/large/best_mIoU_iter_24000.pth'
9
+ checkpoint_teacher_m = '/nas/datasets/lzy/RS-ChangeDetection/checkpoints_distill/STANet/teacher_ckpt/medium/best_mIoU_iter_28000.pth'
10
+ checkpoint_teacher_s = '/nas/datasets/lzy/RS-ChangeDetection/checkpoints_distill/STANet/teacher_ckpt/small/best_mIoU_iter_172000.pth'
11
+
12
+ model = dict(
13
+ # student
14
+ init_cfg=dict(type='Pretrained', checkpoint=checkpoint_student),
15
+ # teacher large
16
+ init_cfg_t_l = dict(type='Pretrained', checkpoint=checkpoint_teacher_l),
17
+ # teacher medium
18
+ init_cfg_t_m = dict(type='Pretrained', checkpoint=checkpoint_teacher_m),
19
+ # teacher small
20
+ init_cfg_t_s = dict(type='Pretrained', checkpoint=checkpoint_teacher_s),
21
+
22
+ decode_head=dict(sa_mode='None'),
23
+ test_cfg=dict(mode='slide', crop_size=crop_size, stride=(crop_size[0]//2, crop_size[1]//2)),
24
+ )
25
+
26
+ param_scheduler = [
27
+ dict(
28
+ type='LinearLR', start_factor=1e-6, by_epoch=False, begin=0, end=1000),
29
+ dict(
30
+ type='PolyLR',
31
+ power=1.0,
32
+ begin=1000,
33
+ end=100000,
34
+ eta_min=0.0,
35
+ by_epoch=False,
36
+ )
37
+ ]
38
+
39
+ # training schedule for 100k
40
+ train_cfg = dict(type='IterBasedTrainLoop', max_iters=100000, val_interval=1000)
41
+ val_cfg = dict(type='ValLoop')
42
+ test_cfg = dict(type='TestLoop')
43
+ default_hooks = dict(
44
+ timer=dict(type='IterTimerHook'),
45
+ logger=dict(type='LoggerHook', interval=100, log_metric_by_epoch=False),
46
+ param_scheduler=dict(type='ParamSchedulerHook'),
47
+ checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=1000,
48
+ save_best='mIoU'),
49
+ sampler_seed=dict(type='DistSamplerSeedHook'),
50
+ visualization=dict(type='CDVisualizationHook', interval=1,
51
+ img_shape=(512, 512, 3)))
STANet/initial/best_mIoU_iter_130000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:377b9b8ab2e7b1bdeb87e33f8271b3f748353e598feec8ec6349d4d5a8cb47a7
3
+ size 65487473
STANet/initial/config.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ _base_ = [
2
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/_base_/models/stanet_r18.py',
3
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/common/standard_512x512_200k_cgwx.py']
4
+
5
+ crop_size = (512, 512)
6
+ model = dict(
7
+ decode_head=dict(sa_mode='None'),
8
+ test_cfg=dict(mode='slide', crop_size=crop_size, stride=(crop_size[0]//2, crop_size[1]//2)),
9
+ )
STANet/large/best_mIoU_iter_24000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:08fe19b1b6cee64cddcff6618014002084a006ec00f925988da2037e600d7ade
3
+ size 53818545
STANet/large/config.py ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _base_ = [
2
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/_base_/models/stanet_r18.py',
3
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/common/train_large_512x512_100k_cgwx.py']
4
+
5
+ crop_size = (512, 512)
6
+
7
+ model = dict(
8
+ decode_head=dict(sa_mode='None'),
9
+ test_cfg=dict(mode='slide', crop_size=crop_size, stride=(crop_size[0]//2, crop_size[1]//2)),
10
+ )
11
+
12
+ param_scheduler = [
13
+ dict(
14
+ type='LinearLR', start_factor=1e-6, by_epoch=False, begin=0, end=1000),
15
+ dict(
16
+ type='PolyLR',
17
+ power=1.0,
18
+ begin=1000,
19
+ end=200000,
20
+ eta_min=0.0,
21
+ by_epoch=False,
22
+ )
23
+ ]
24
+ # training schedule for 100k
25
+ train_cfg = dict(type='IterBasedTrainLoop', max_iters=200000, val_interval=1000)
26
+ val_cfg = dict(type='ValLoop')
27
+ test_cfg = dict(type='TestLoop')
28
+ default_hooks = dict(
29
+ timer=dict(type='IterTimerHook'),
30
+ logger=dict(type='LoggerHook', interval=100, log_metric_by_epoch=False),
31
+ param_scheduler=dict(type='ParamSchedulerHook'),
32
+ checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=1000,
33
+ save_best='mIoU'),
34
+ sampler_seed=dict(type='DistSamplerSeedHook'),
35
+ visualization=dict(type='CDVisualizationHook', interval=1,
36
+ img_shape=(512, 512, 3)))
STANet/medium/best_mIoU_iter_28000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bcdaf3ed0d8202bf0a853bb84f144925833799381e2a25d1059bc85cf0e14846
3
+ size 54268657
STANet/medium/config.py ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _base_ = [
2
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/_base_/models/stanet_r18.py',
3
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/common/train_medium_512x512_100k_cgwx.py']
4
+
5
+ crop_size = (512, 512)
6
+
7
+ model = dict(
8
+ decode_head=dict(sa_mode='None'),
9
+ test_cfg=dict(mode='slide', crop_size=crop_size, stride=(crop_size[0]//2, crop_size[1]//2)),
10
+ )
11
+
12
+ param_scheduler = [
13
+ dict(
14
+ type='LinearLR', start_factor=1e-6, by_epoch=False, begin=0, end=1000),
15
+ dict(
16
+ type='PolyLR',
17
+ power=1.0,
18
+ begin=1000,
19
+ end=200000,
20
+ eta_min=0.0,
21
+ by_epoch=False,
22
+ )
23
+ ]
24
+ # training schedule for 100k
25
+ train_cfg = dict(type='IterBasedTrainLoop', max_iters=200000, val_interval=1000)
26
+ val_cfg = dict(type='ValLoop')
27
+ test_cfg = dict(type='TestLoop')
28
+ default_hooks = dict(
29
+ timer=dict(type='IterTimerHook'),
30
+ logger=dict(type='LoggerHook', interval=100, log_metric_by_epoch=False),
31
+ param_scheduler=dict(type='ParamSchedulerHook'),
32
+ checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=1000,
33
+ save_best='mIoU'),
34
+ sampler_seed=dict(type='DistSamplerSeedHook'),
35
+ visualization=dict(type='CDVisualizationHook', interval=1,
36
+ img_shape=(512, 512, 3)))
STANet/small/best_mIoU_iter_172000.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b4ac8a22ab0982ba959ef6409f7becda8c1f609411f90ece2ddaad379cbe52f4
3
+ size 70357297
STANet/small/config.py ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _base_ = [
2
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/_base_/models/stanet_r18.py',
3
+ '/home/liuziyuan/proj/RS-CD/rs-cd-cgwx/configs/common/train_small_512x512_100k_cgwx.py']
4
+
5
+ crop_size = (512, 512)
6
+
7
+ model = dict(
8
+ decode_head=dict(sa_mode='None'),
9
+ test_cfg=dict(mode='slide', crop_size=crop_size, stride=(crop_size[0]//2, crop_size[1]//2)),
10
+ )
11
+
12
+ param_scheduler = [
13
+ dict(
14
+ type='LinearLR', start_factor=1e-6, by_epoch=False, begin=0, end=1000),
15
+ dict(
16
+ type='PolyLR',
17
+ power=1.0,
18
+ begin=1000,
19
+ end=200000,
20
+ eta_min=0.0,
21
+ by_epoch=False,
22
+ )
23
+ ]
24
+ # training schedule for 100k
25
+ train_cfg = dict(type='IterBasedTrainLoop', max_iters=200000, val_interval=1000)
26
+ val_cfg = dict(type='ValLoop')
27
+ test_cfg = dict(type='TestLoop')
28
+ default_hooks = dict(
29
+ timer=dict(type='IterTimerHook'),
30
+ logger=dict(type='LoggerHook', interval=100, log_metric_by_epoch=False),
31
+ param_scheduler=dict(type='ParamSchedulerHook'),
32
+ checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=1000,
33
+ save_best='mIoU'),
34
+ sampler_seed=dict(type='DistSamplerSeedHook'),
35
+ visualization=dict(type='CDVisualizationHook', interval=1,
36
+ img_shape=(512, 512, 3)))