Upload 50 files
Browse files- HANet/distill/best_mIoU_iter_90000.pth +3 -0
- HANet/distill/config.py +51 -0
- HANet/initial/best_mIoU_iter_195000.pth +3 -0
- HANet/initial/config.py +3 -0
- HANet/large/best_mIoU_iter_122000.pth +3 -0
- HANet/large/config.py +29 -0
- HANet/medium/best_mIoU_iter_74000.pth +3 -0
- HANet/medium/config.py +29 -0
- HANet/small/best_mIoU_iter_158000.pth +3 -0
- HANet/small/config.py +29 -0
- IFN/distill/best_mIoU_iter_84000.pth +3 -0
- IFN/distill/config.py +51 -0
- IFN/initial/best_mIoU_iter_110000.pth +3 -0
- IFN/initial/config.py +3 -0
- IFN/large/best_mIoU_iter_173000.pth +3 -0
- IFN/large/config.py +29 -0
- IFN/medium/best_mIoU_iter_7000.pth +3 -0
- IFN/medium/config.py +29 -0
- IFN/small/best_mIoU_iter_145000.pth +3 -0
- IFN/small/config.py +29 -0
- LightCDNet/distill/best_mIoU_iter_168000.pth +3 -0
- LightCDNet/distill/config.py +65 -0
- LightCDNet/initial/best_mIoU_iter_195000.pth +3 -0
- LightCDNet/initial/config.py +19 -0
- LightCDNet/large/best_mIoU_iter_185000.pth +3 -0
- LightCDNet/large/config.py +50 -0
- LightCDNet/medium/best_mIoU_iter_199000.pth +3 -0
- LightCDNet/medium/config.py +45 -0
- LightCDNet/small/best_mIoU_iter_193000.pth +3 -0
- LightCDNet/small/config.py +45 -0
- SNUNet/distill/best_mIoU_iter_88000.pth +3 -0
- SNUNet/distill/config.py +51 -0
- SNUNet/initial/best_mIoU_iter_160000.pth +3 -0
- SNUNet/initial/config.py +9 -0
- SNUNet/large/best_mIoU_iter_62000.pth +3 -0
- SNUNet/large/config.py +34 -0
- SNUNet/medium/best_mIoU_iter_20000.pth +3 -0
- SNUNet/medium/config.py +34 -0
- SNUNet/small/best_mIoU_iter_177000.pth +3 -0
- SNUNet/small/config.py +34 -0
- STANet/distill/best_mIoU_iter_46000.pth +3 -0
- STANet/distill/config.py +51 -0
- STANet/initial/best_mIoU_iter_130000.pth +3 -0
- STANet/initial/config.py +9 -0
- STANet/large/best_mIoU_iter_24000.pth +3 -0
- STANet/large/config.py +36 -0
- STANet/medium/best_mIoU_iter_28000.pth +3 -0
- STANet/medium/config.py +36 -0
- STANet/small/best_mIoU_iter_172000.pth +3 -0
- 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)))
|