parent
2b17dbf4e4
commit
3d942f9868
37 changed files with 28 additions and 588 deletions
@ -0,0 +1,8 @@ |
|||||||
|
# Basic configurations of BIT |
||||||
|
|
||||||
|
_base_: ../_base_/airchange.yaml |
||||||
|
|
||||||
|
save_dir: ./test_tipc/output/cd/bit/ |
||||||
|
|
||||||
|
model: !Node |
||||||
|
type: BIT |
@ -1,8 +0,0 @@ |
|||||||
# Configurations of CDNet with AirChange dataset |
|
||||||
|
|
||||||
_base_: ../_base_/airchange.yaml |
|
||||||
|
|
||||||
save_dir: ./test_tipc/output/cd/cdnet/ |
|
||||||
|
|
||||||
model: !Node |
|
||||||
type: CDNet |
|
@ -1,8 +0,0 @@ |
|||||||
# Configurations of cdnet with LEVIR-CD dataset |
|
||||||
|
|
||||||
_base_: ../_base_/levircd.yaml |
|
||||||
|
|
||||||
save_dir: ./test_tipc/output/cd/cdnet/ |
|
||||||
|
|
||||||
model: !Node |
|
||||||
type: CDNet |
|
@ -1,53 +0,0 @@ |
|||||||
===========================train_params=========================== |
|
||||||
model_name:cd:cdnet |
|
||||||
python:python |
|
||||||
gpu_list:0|0,1 |
|
||||||
use_gpu:null|null |
|
||||||
--precision:null |
|
||||||
--num_epochs:lite_train_lite_infer=5|lite_train_whole_infer=5|whole_train_whole_infer=10 |
|
||||||
--save_dir:adaptive |
|
||||||
--train_batch_size:lite_train_lite_infer=4|lite_train_whole_infer=4|whole_train_whole_infer=8 |
|
||||||
--model_path:null |
|
||||||
--config:lite_train_lite_infer=./test_tipc/configs/cd/cdnet/cdnet_airchange.yaml|lite_train_whole_infer=./test_tipc/configs/cd/cdnet/cdnet_airchange.yaml|whole_train_whole_infer=./test_tipc/configs/cd/cdnet/cdnet_levircd.yaml |
|
||||||
train_model_name:best_model |
|
||||||
null:null |
|
||||||
## |
|
||||||
trainer:norm |
|
||||||
norm_train:test_tipc/run_task.py train cd |
|
||||||
pact_train:null |
|
||||||
fpgm_train:null |
|
||||||
distill_train:null |
|
||||||
null:null |
|
||||||
null:null |
|
||||||
## |
|
||||||
===========================eval_params=========================== |
|
||||||
eval:null |
|
||||||
null:null |
|
||||||
## |
|
||||||
===========================export_params=========================== |
|
||||||
--save_dir:adaptive |
|
||||||
--model_dir:adaptive |
|
||||||
--fixed_input_shape:[-1,3,256,256] |
|
||||||
norm_export:deploy/export/export_model.py |
|
||||||
quant_export:null |
|
||||||
fpgm_export:null |
|
||||||
distill_export:null |
|
||||||
export1:null |
|
||||||
export2:null |
|
||||||
===========================infer_params=========================== |
|
||||||
infer_model:null |
|
||||||
infer_export:null |
|
||||||
infer_quant:False |
|
||||||
inference:test_tipc/infer.py |
|
||||||
--device:cpu|gpu |
|
||||||
--enable_mkldnn:True |
|
||||||
--cpu_threads:6 |
|
||||||
--batch_size:1 |
|
||||||
--use_trt:False |
|
||||||
--precision:fp32 |
|
||||||
--model_dir:null |
|
||||||
--config:null |
|
||||||
--save_log_path:null |
|
||||||
--benchmark:True |
|
||||||
--model_name:cdnet |
|
||||||
null:null |
|
@ -1,4 +1,4 @@ |
|||||||
# Configurations of ChangeFormer with AirChange dataset |
# Basic configurations of ChangeFormer |
||||||
|
|
||||||
_base_: ../_base_/airchange.yaml |
_base_: ../_base_/airchange.yaml |
||||||
|
|
@ -1,8 +0,0 @@ |
|||||||
# Configurations of ChangeFormer with LEVIR-CD dataset |
|
||||||
|
|
||||||
_base_: ../_base_/levircd.yaml |
|
||||||
|
|
||||||
save_dir: ./test_tipc/output/cd/changeformer/ |
|
||||||
|
|
||||||
model: !Node |
|
||||||
type: ChangeFormer |
|
@ -1,8 +0,0 @@ |
|||||||
# Configurations of DSAMNet with AirChange dataset |
|
||||||
|
|
||||||
_base_: ../_base_/airchange.yaml |
|
||||||
|
|
||||||
save_dir: ./test_tipc/output/cd/dsamnet/ |
|
||||||
|
|
||||||
model: !Node |
|
||||||
type: DSAMNet |
|
@ -1,8 +0,0 @@ |
|||||||
# Configurations of DSAMNet with LEVIR-CD dataset |
|
||||||
|
|
||||||
_base_: ../_base_/levircd.yaml |
|
||||||
|
|
||||||
save_dir: ./test_tipc/output/cd/dsamnet/ |
|
||||||
|
|
||||||
model: !Node |
|
||||||
type: DSAMNet |
|
@ -1,53 +0,0 @@ |
|||||||
===========================train_params=========================== |
|
||||||
model_name:cd:dsamnet |
|
||||||
python:python |
|
||||||
gpu_list:0|0,1 |
|
||||||
use_gpu:null|null |
|
||||||
--precision:null |
|
||||||
--num_epochs:lite_train_lite_infer=5|lite_train_whole_infer=5|whole_train_whole_infer=10 |
|
||||||
--save_dir:adaptive |
|
||||||
--train_batch_size:lite_train_lite_infer=4|lite_train_whole_infer=4|whole_train_whole_infer=8 |
|
||||||
--model_path:null |
|
||||||
--config:lite_train_lite_infer=./test_tipc/configs/cd/dsamnet/dsamnet_airchange.yaml|lite_train_whole_infer=./test_tipc/configs/cd/dsamnet/dsamnet_airchange.yaml|whole_train_whole_infer=./test_tipc/configs/cd/dsamnet/dsamnet_levircd.yaml |
|
||||||
train_model_name:best_model |
|
||||||
null:null |
|
||||||
## |
|
||||||
trainer:norm |
|
||||||
norm_train:test_tipc/run_task.py train cd |
|
||||||
pact_train:null |
|
||||||
fpgm_train:null |
|
||||||
distill_train:null |
|
||||||
null:null |
|
||||||
null:null |
|
||||||
## |
|
||||||
===========================eval_params=========================== |
|
||||||
eval:null |
|
||||||
null:null |
|
||||||
## |
|
||||||
===========================export_params=========================== |
|
||||||
--save_dir:adaptive |
|
||||||
--model_dir:adaptive |
|
||||||
--fixed_input_shape:[-1,3,256,256] |
|
||||||
norm_export:deploy/export/export_model.py |
|
||||||
quant_export:null |
|
||||||
fpgm_export:null |
|
||||||
distill_export:null |
|
||||||
export1:null |
|
||||||
export2:null |
|
||||||
===========================infer_params=========================== |
|
||||||
infer_model:null |
|
||||||
infer_export:null |
|
||||||
infer_quant:False |
|
||||||
inference:test_tipc/infer.py |
|
||||||
--device:cpu|gpu |
|
||||||
--enable_mkldnn:True |
|
||||||
--cpu_threads:6 |
|
||||||
--batch_size:1 |
|
||||||
--use_trt:False |
|
||||||
--precision:fp32 |
|
||||||
--model_dir:null |
|
||||||
--config:null |
|
||||||
--save_log_path:null |
|
||||||
--benchmark:True |
|
||||||
--model_name:dsamnet |
|
||||||
null:null |
|
@ -1,8 +0,0 @@ |
|||||||
# Configurations of DSIFN with AirChange dataset |
|
||||||
|
|
||||||
_base_: ../_base_/airchange.yaml |
|
||||||
|
|
||||||
save_dir: ./test_tipc/output/cd/dsifn/ |
|
||||||
|
|
||||||
model: !Node |
|
||||||
type: DSIFN |
|
@ -1,8 +0,0 @@ |
|||||||
# Configurations of DSIFN with LEVIR-CD dataset |
|
||||||
|
|
||||||
_base_: ../_base_/levircd.yaml |
|
||||||
|
|
||||||
save_dir: ./test_tipc/output/cd/dsifn/ |
|
||||||
|
|
||||||
model: !Node |
|
||||||
type: DSIFN |
|
@ -1,53 +0,0 @@ |
|||||||
===========================train_params=========================== |
|
||||||
model_name:cd:dsifn |
|
||||||
python:python |
|
||||||
gpu_list:0|0,1 |
|
||||||
use_gpu:null|null |
|
||||||
--precision:null |
|
||||||
--num_epochs:lite_train_lite_infer=5|lite_train_whole_infer=5|whole_train_whole_infer=10 |
|
||||||
--save_dir:adaptive |
|
||||||
--train_batch_size:lite_train_lite_infer=4|lite_train_whole_infer=4|whole_train_whole_infer=8 |
|
||||||
--model_path:null |
|
||||||
--config:lite_train_lite_infer=./test_tipc/configs/cd/dsifn/dsifn_airchange.yaml|lite_train_whole_infer=./test_tipc/configs/cd/dsifn/dsifn_airchange.yaml|whole_train_whole_infer=./test_tipc/configs/cd/dsifn/dsifn_levircd.yaml |
|
||||||
train_model_name:best_model |
|
||||||
null:null |
|
||||||
## |
|
||||||
trainer:norm |
|
||||||
norm_train:test_tipc/run_task.py train cd |
|
||||||
pact_train:null |
|
||||||
fpgm_train:null |
|
||||||
distill_train:null |
|
||||||
null:null |
|
||||||
null:null |
|
||||||
## |
|
||||||
===========================eval_params=========================== |
|
||||||
eval:null |
|
||||||
null:null |
|
||||||
## |
|
||||||
===========================export_params=========================== |
|
||||||
--save_dir:adaptive |
|
||||||
--model_dir:adaptive |
|
||||||
--fixed_input_shape:[-1,3,256,256] |
|
||||||
norm_export:deploy/export/export_model.py |
|
||||||
quant_export:null |
|
||||||
fpgm_export:null |
|
||||||
distill_export:null |
|
||||||
export1:null |
|
||||||
export2:null |
|
||||||
===========================infer_params=========================== |
|
||||||
infer_model:null |
|
||||||
infer_export:null |
|
||||||
infer_quant:False |
|
||||||
inference:test_tipc/infer.py |
|
||||||
--device:cpu|gpu |
|
||||||
--enable_mkldnn:True |
|
||||||
--cpu_threads:6 |
|
||||||
--batch_size:1 |
|
||||||
--use_trt:False |
|
||||||
--precision:fp32 |
|
||||||
--model_dir:null |
|
||||||
--config:null |
|
||||||
--save_log_path:null |
|
||||||
--benchmark:True |
|
||||||
--model_name:dsifn |
|
||||||
null:null |
|
@ -1,8 +0,0 @@ |
|||||||
# Configurations of FC-EF with AirChange dataset |
|
||||||
|
|
||||||
_base_: ../_base_/airchange.yaml |
|
||||||
|
|
||||||
save_dir: ./test_tipc/output/cd/fc_ef/ |
|
||||||
|
|
||||||
model: !Node |
|
||||||
type: FCEarlyFusion |
|
@ -1,8 +0,0 @@ |
|||||||
# Configurations of FC-EF with LEVIR-CD dataset |
|
||||||
|
|
||||||
_base_: ../_base_/levircd.yaml |
|
||||||
|
|
||||||
save_dir: ./test_tipc/output/cd/fc_ef/ |
|
||||||
|
|
||||||
model: !Node |
|
||||||
type: FCEarlyFusion |
|
@ -1,53 +0,0 @@ |
|||||||
===========================train_params=========================== |
|
||||||
model_name:cd:fc_ef |
|
||||||
python:python |
|
||||||
gpu_list:0|0,1 |
|
||||||
use_gpu:null|null |
|
||||||
--precision:null |
|
||||||
--num_epochs:lite_train_lite_infer=5|lite_train_whole_infer=5|whole_train_whole_infer=10 |
|
||||||
--save_dir:adaptive |
|
||||||
--train_batch_size:lite_train_lite_infer=4|lite_train_whole_infer=4|whole_train_whole_infer=8 |
|
||||||
--model_path:null |
|
||||||
--config:lite_train_lite_infer=./test_tipc/configs/cd/fc_ef/fc_ef_airchange.yaml|lite_train_whole_infer=./test_tipc/configs/cd/fc_ef/fc_ef_airchange.yaml|whole_train_whole_infer=./test_tipc/configs/cd/fc_ef/fc_ef_levircd.yaml |
|
||||||
train_model_name:best_model |
|
||||||
null:null |
|
||||||
## |
|
||||||
trainer:norm |
|
||||||
norm_train:test_tipc/run_task.py train cd |
|
||||||
pact_train:null |
|
||||||
fpgm_train:null |
|
||||||
distill_train:null |
|
||||||
null:null |
|
||||||
null:null |
|
||||||
## |
|
||||||
===========================eval_params=========================== |
|
||||||
eval:null |
|
||||||
null:null |
|
||||||
## |
|
||||||
===========================export_params=========================== |
|
||||||
--save_dir:adaptive |
|
||||||
--model_dir:adaptive |
|
||||||
--fixed_input_shape:[-1,3,256,256] |
|
||||||
norm_export:deploy/export/export_model.py |
|
||||||
quant_export:null |
|
||||||
fpgm_export:null |
|
||||||
distill_export:null |
|
||||||
export1:null |
|
||||||
export2:null |
|
||||||
===========================infer_params=========================== |
|
||||||
infer_model:null |
|
||||||
infer_export:null |
|
||||||
infer_quant:False |
|
||||||
inference:test_tipc/infer.py |
|
||||||
--device:cpu|gpu |
|
||||||
--enable_mkldnn:True |
|
||||||
--cpu_threads:6 |
|
||||||
--batch_size:1 |
|
||||||
--use_trt:False |
|
||||||
--precision:fp32 |
|
||||||
--model_dir:null |
|
||||||
--config:null |
|
||||||
--save_log_path:null |
|
||||||
--benchmark:True |
|
||||||
--model_name:fc_ef |
|
||||||
null:null |
|
@ -1,8 +0,0 @@ |
|||||||
# Configurations of FC-Siam-conc with AirChange dataset |
|
||||||
|
|
||||||
_base_: ../_base_/airchange.yaml |
|
||||||
|
|
||||||
save_dir: ./test_tipc/output/cd/fc_siam_conc/ |
|
||||||
|
|
||||||
model: !Node |
|
||||||
type: FCSiamConc |
|
@ -1,8 +0,0 @@ |
|||||||
# Configurations of FC-Siam-conc with LEVIR-CD dataset |
|
||||||
|
|
||||||
_base_: ../_base_/levircd.yaml |
|
||||||
|
|
||||||
save_dir: ./test_tipc/output/cd/fc_siam_conc/ |
|
||||||
|
|
||||||
model: !Node |
|
||||||
type: FCSiamConc |
|
@ -1,53 +0,0 @@ |
|||||||
===========================train_params=========================== |
|
||||||
model_name:cd:fc_siam_conc |
|
||||||
python:python |
|
||||||
gpu_list:0|0,1 |
|
||||||
use_gpu:null|null |
|
||||||
--precision:null |
|
||||||
--num_epochs:lite_train_lite_infer=5|lite_train_whole_infer=5|whole_train_whole_infer=10 |
|
||||||
--save_dir:adaptive |
|
||||||
--train_batch_size:lite_train_lite_infer=4|lite_train_whole_infer=4|whole_train_whole_infer=8 |
|
||||||
--model_path:null |
|
||||||
--config:lite_train_lite_infer=./test_tipc/configs/cd/fc_siam_conc/fc_siam_conc_airchange.yaml|lite_train_whole_infer=./test_tipc/configs/cd/fc_siam_conc/fc_siam_conc_airchange.yaml|whole_train_whole_infer=./test_tipc/configs/cd/fc_siam_conc/fc_siam_conc_levircd.yaml |
|
||||||
train_model_name:best_model |
|
||||||
null:null |
|
||||||
## |
|
||||||
trainer:norm |
|
||||||
norm_train:test_tipc/run_task.py train cd |
|
||||||
pact_train:null |
|
||||||
fpgm_train:null |
|
||||||
distill_train:null |
|
||||||
null:null |
|
||||||
null:null |
|
||||||
## |
|
||||||
===========================eval_params=========================== |
|
||||||
eval:null |
|
||||||
null:null |
|
||||||
## |
|
||||||
===========================export_params=========================== |
|
||||||
--save_dir:adaptive |
|
||||||
--model_dir:adaptive |
|
||||||
--fixed_input_shape:[-1,3,256,256] |
|
||||||
norm_export:deploy/export/export_model.py |
|
||||||
quant_export:null |
|
||||||
fpgm_export:null |
|
||||||
distill_export:null |
|
||||||
export1:null |
|
||||||
export2:null |
|
||||||
===========================infer_params=========================== |
|
||||||
infer_model:null |
|
||||||
infer_export:null |
|
||||||
infer_quant:False |
|
||||||
inference:test_tipc/infer.py |
|
||||||
--device:cpu|gpu |
|
||||||
--enable_mkldnn:True |
|
||||||
--cpu_threads:6 |
|
||||||
--batch_size:1 |
|
||||||
--use_trt:False |
|
||||||
--precision:fp32 |
|
||||||
--model_dir:null |
|
||||||
--config:null |
|
||||||
--save_log_path:null |
|
||||||
--benchmark:True |
|
||||||
--model_name:fc_siam_conc |
|
||||||
null:null |
|
@ -1,8 +0,0 @@ |
|||||||
# Configurations of FC-Siam-diff with AirChange dataset |
|
||||||
|
|
||||||
_base_: ../_base_/airchange.yaml |
|
||||||
|
|
||||||
save_dir: ./test_tipc/output/cd/fc_siam_diff/ |
|
||||||
|
|
||||||
model: !Node |
|
||||||
type: FCSiamDiff |
|
@ -1,8 +0,0 @@ |
|||||||
# Configurations of FC-Siam-diff with LEVIR-CD dataset |
|
||||||
|
|
||||||
_base_: ../_base_/levircd.yaml |
|
||||||
|
|
||||||
save_dir: ./test_tipc/output/cd/fc_siam_diff/ |
|
||||||
|
|
||||||
model: !Node |
|
||||||
type: FCSiamDiff |
|
@ -1,53 +0,0 @@ |
|||||||
===========================train_params=========================== |
|
||||||
model_name:cd:fc_siam_diff |
|
||||||
python:python |
|
||||||
gpu_list:0|0,1 |
|
||||||
use_gpu:null|null |
|
||||||
--precision:null |
|
||||||
--num_epochs:lite_train_lite_infer=5|lite_train_whole_infer=5|whole_train_whole_infer=10 |
|
||||||
--save_dir:adaptive |
|
||||||
--train_batch_size:lite_train_lite_infer=4|lite_train_whole_infer=4|whole_train_whole_infer=8 |
|
||||||
--model_path:null |
|
||||||
--config:lite_train_lite_infer=./test_tipc/configs/cd/fc_siam_diff/fc_siam_diff_airchange.yaml|lite_train_whole_infer=./test_tipc/configs/cd/fc_siam_diff/fc_siam_diff_airchange.yaml|whole_train_whole_infer=./test_tipc/configs/cd/fc_siam_diff/fc_siam_diff_levircd.yaml |
|
||||||
train_model_name:best_model |
|
||||||
null:null |
|
||||||
## |
|
||||||
trainer:norm |
|
||||||
norm_train:test_tipc/run_task.py train cd |
|
||||||
pact_train:null |
|
||||||
fpgm_train:null |
|
||||||
distill_train:null |
|
||||||
null:null |
|
||||||
null:null |
|
||||||
## |
|
||||||
===========================eval_params=========================== |
|
||||||
eval:null |
|
||||||
null:null |
|
||||||
## |
|
||||||
===========================export_params=========================== |
|
||||||
--save_dir:adaptive |
|
||||||
--model_dir:adaptive |
|
||||||
--fixed_input_shape:[-1,3,256,256] |
|
||||||
norm_export:deploy/export/export_model.py |
|
||||||
quant_export:null |
|
||||||
fpgm_export:null |
|
||||||
distill_export:null |
|
||||||
export1:null |
|
||||||
export2:null |
|
||||||
===========================infer_params=========================== |
|
||||||
infer_model:null |
|
||||||
infer_export:null |
|
||||||
infer_quant:False |
|
||||||
inference:test_tipc/infer.py |
|
||||||
--device:cpu|gpu |
|
||||||
--enable_mkldnn:True |
|
||||||
--cpu_threads:6 |
|
||||||
--batch_size:1 |
|
||||||
--use_trt:False |
|
||||||
--precision:fp32 |
|
||||||
--model_dir:null |
|
||||||
--config:null |
|
||||||
--save_log_path:null |
|
||||||
--benchmark:True |
|
||||||
--model_name:fc_siam_diff |
|
||||||
null:null |
|
@ -1,8 +0,0 @@ |
|||||||
# Configurations of SNUNet with AirChange dataset |
|
||||||
|
|
||||||
_base_: ../_base_/airchange.yaml |
|
||||||
|
|
||||||
save_dir: ./test_tipc/output/cd/snunet/ |
|
||||||
|
|
||||||
model: !Node |
|
||||||
type: SNUNet |
|
@ -1,8 +0,0 @@ |
|||||||
# Configurations of SNUNet with LEVIR-CD dataset |
|
||||||
|
|
||||||
_base_: ../_base_/levircd.yaml |
|
||||||
|
|
||||||
save_dir: ./test_tipc/output/cd/snunet/ |
|
||||||
|
|
||||||
model: !Node |
|
||||||
type: SNUNet |
|
@ -1,53 +0,0 @@ |
|||||||
===========================train_params=========================== |
|
||||||
model_name:cd:snunet |
|
||||||
python:python |
|
||||||
gpu_list:0|0,1 |
|
||||||
use_gpu:null|null |
|
||||||
--precision:null |
|
||||||
--num_epochs:lite_train_lite_infer=5|lite_train_whole_infer=5|whole_train_whole_infer=10 |
|
||||||
--save_dir:adaptive |
|
||||||
--train_batch_size:lite_train_lite_infer=4|lite_train_whole_infer=4|whole_train_whole_infer=8 |
|
||||||
--model_path:null |
|
||||||
--config:lite_train_lite_infer=./test_tipc/configs/cd/snunet/snunet_airchange.yaml|lite_train_whole_infer=./test_tipc/configs/cd/snunet/snunet_airchange.yaml|whole_train_whole_infer=./test_tipc/configs/cd/snunet/snunet_levircd.yaml |
|
||||||
train_model_name:best_model |
|
||||||
null:null |
|
||||||
## |
|
||||||
trainer:norm |
|
||||||
norm_train:test_tipc/run_task.py train cd |
|
||||||
pact_train:null |
|
||||||
fpgm_train:null |
|
||||||
distill_train:null |
|
||||||
null:null |
|
||||||
null:null |
|
||||||
## |
|
||||||
===========================eval_params=========================== |
|
||||||
eval:null |
|
||||||
null:null |
|
||||||
## |
|
||||||
===========================export_params=========================== |
|
||||||
--save_dir:adaptive |
|
||||||
--model_dir:adaptive |
|
||||||
--fixed_input_shape:[-1,3,256,256] |
|
||||||
norm_export:deploy/export/export_model.py |
|
||||||
quant_export:null |
|
||||||
fpgm_export:null |
|
||||||
distill_export:null |
|
||||||
export1:null |
|
||||||
export2:null |
|
||||||
===========================infer_params=========================== |
|
||||||
infer_model:null |
|
||||||
infer_export:null |
|
||||||
infer_quant:False |
|
||||||
inference:test_tipc/infer.py |
|
||||||
--device:cpu|gpu |
|
||||||
--enable_mkldnn:True |
|
||||||
--cpu_threads:6 |
|
||||||
--batch_size:1 |
|
||||||
--use_trt:False |
|
||||||
--precision:fp32 |
|
||||||
--model_dir:null |
|
||||||
--config:null |
|
||||||
--save_log_path:null |
|
||||||
--benchmark:True |
|
||||||
--model_name:snunet |
|
||||||
null:null |
|
@ -1,8 +0,0 @@ |
|||||||
# Configurations of STANet with AirChange dataset |
|
||||||
|
|
||||||
_base_: ../_base_/airchange.yaml |
|
||||||
|
|
||||||
save_dir: ./test_tipc/output/cd/stanet/ |
|
||||||
|
|
||||||
model: !Node |
|
||||||
type: STANet |
|
@ -1,8 +0,0 @@ |
|||||||
# Configurations of STANet with LEVIR-CD dataset |
|
||||||
|
|
||||||
_base_: ../_base_/levircd.yaml |
|
||||||
|
|
||||||
save_dir: ./test_tipc/output/cd/stanet/ |
|
||||||
|
|
||||||
model: !Node |
|
||||||
type: STANet |
|
@ -1,53 +0,0 @@ |
|||||||
===========================train_params=========================== |
|
||||||
model_name:cd:stanet |
|
||||||
python:python |
|
||||||
gpu_list:0|0,1 |
|
||||||
use_gpu:null|null |
|
||||||
--precision:null |
|
||||||
--num_epochs:lite_train_lite_infer=5|lite_train_whole_infer=5|whole_train_whole_infer=10 |
|
||||||
--save_dir:adaptive |
|
||||||
--train_batch_size:lite_train_lite_infer=4|lite_train_whole_infer=4|whole_train_whole_infer=8 |
|
||||||
--model_path:null |
|
||||||
--config:lite_train_lite_infer=./test_tipc/configs/cd/stanet/stanet_airchange.yaml|lite_train_whole_infer=./test_tipc/configs/cd/stanet/stanet_airchange.yaml|whole_train_whole_infer=./test_tipc/configs/cd/stanet/stanet_levircd.yaml |
|
||||||
train_model_name:best_model |
|
||||||
null:null |
|
||||||
## |
|
||||||
trainer:norm |
|
||||||
norm_train:test_tipc/run_task.py train cd |
|
||||||
pact_train:null |
|
||||||
fpgm_train:null |
|
||||||
distill_train:null |
|
||||||
null:null |
|
||||||
null:null |
|
||||||
## |
|
||||||
===========================eval_params=========================== |
|
||||||
eval:null |
|
||||||
null:null |
|
||||||
## |
|
||||||
===========================export_params=========================== |
|
||||||
--save_dir:adaptive |
|
||||||
--model_dir:adaptive |
|
||||||
--fixed_input_shape:[-1,3,256,256] |
|
||||||
norm_export:deploy/export/export_model.py |
|
||||||
quant_export:null |
|
||||||
fpgm_export:null |
|
||||||
distill_export:null |
|
||||||
export1:null |
|
||||||
export2:null |
|
||||||
===========================infer_params=========================== |
|
||||||
infer_model:null |
|
||||||
infer_export:null |
|
||||||
infer_quant:False |
|
||||||
inference:test_tipc/infer.py |
|
||||||
--device:cpu|gpu |
|
||||||
--enable_mkldnn:True |
|
||||||
--cpu_threads:6 |
|
||||||
--batch_size:1 |
|
||||||
--use_trt:False |
|
||||||
--precision:fp32 |
|
||||||
--model_dir:null |
|
||||||
--config:null |
|
||||||
--save_log_path:null |
|
||||||
--benchmark:True |
|
||||||
--model_name:stanet |
|
||||||
null:null |
|
@ -1,4 +1,4 @@ |
|||||||
# Configurations of HRNet with UCMerced dataset |
# Basic configurations of HRNet |
||||||
|
|
||||||
_base_: ../_base_/ucmerced.yaml |
_base_: ../_base_/ucmerced.yaml |
||||||
|
|
Loading…
Reference in new issue