FunSR / tools /paper_vis_tools /get_feature_map_vis.py
KyanChen's picture
add
02c5426
import os
# import sys
# sys.path.append(sys.path[0]+'/../../')
exp = 'EXP20221219_1'
model_name = 'FunSR-RDN' # bicubic, SRCNN, FSRCNN, LGCNet
dataset_name = 'AID' # UC, AID
for cp in ['epoch-last.pth']:
for scale_ratio in [4.0]:
# os.system(f'CUDA_VISIBLE_DEVICES=2 python test_cnn_sr.py '
# f'--config tools/paper_tools/vis_fixed_scale_UC_INR_diinn_arbrcan_funsr_overnet.yaml '
# f'--model checkpoints/{exp}/{cp} '
# f'--scale_ratio {scale_ratio} '
# f'--save_fig True '
# f'--save_path vis_{model_name}_{dataset_name}_4x_testset '
# f'--cal_metrics True '
# f'--dataset_name {dataset_name}'
# )
os.system(f'CUDA_VISIBLE_DEVICES=2 python test_inr_diinn_arbrcan_sadnarc_funsr_overnet.py '
f'--config tools/paper_tools/vis_fixed_scale_UC_INR_diinn_arbrcan_funsr_overnet.yaml '
f'--model checkpoints/{exp}/{cp} '
f'--scale_ratio {scale_ratio} '
f'--save_fig False '
f'--save_featmap True '
f'--save_path vis_{model_name}_{dataset_name}_4x_testset_featmap '
f'--cal_metrics True '
f'--dataset_name {dataset_name}'
)
# os.system(f'CUDA_VISIBLE_DEVICES=5 python test_inr_liif_metasr_aliif.py '
# f'--config tools/paper_tools/vis_fixed_scale_UC_INR_liif_metasr_aliif.yaml '
# f'--model checkpoints/{exp}/{cp} '
# f'--scale_ratio {scale_ratio} '
# f'--save_fig True '
# f'--save_path vis_{model_name}_{dataset_name}_4x_testset '
# f'--cal_metrics True '
# f'--dataset_name {dataset_name}'
# )
# os.system(f'zip -q -r vis_{model_name}_{dataset_name}_4x_testset_featmap.zip vis_{model_name}_{dataset_name}_4x_testset')
# os.system(f'aws s3 cp vis_{model_name}_{dataset_name}_4x_testset_featmap.zip s3://xhs.bravo/user/kyanchen/tmp/')