search

mindspore-lab / mindediting

Low level vision toolbox based on MindSpore
Apache License 2.0
59 stars 8 forks source link

How to solve KeyError: 'basicvsr' #6

Open Simon-ssp opened 3 months ago

Simon-ssp commented 3 months ago

I just run 'python val.py --config_path ./configs/basicvsr/val.yaml --save_metrics ./output/quality_metrics.json'. But i got : raceback (most recent call last): File "val.py", line 51, in val(final_config, path_args.save_metrics) File "val.py", line 35, in val net, eval_network = create_model_by_name(model_name=cfg.model.name, cfg=cfg) File "/home/ma-user/work/mindediting-master/mindediting/models/init.py", line 491, in create_model_by_name cfg, network_creator = get_model_info(model_name, cfg) File "/home/ma-user/work/mindediting-master/mindediting/models/init.py", line 506, in get_model_info model_creator_name = MODEL_NAME_TO_DEFAULT_INFO[model_name][0] KeyError: 'basicvsr'

Simon-ssp commented 3 months ago

I added "basicvsr" to MODEL_NAME_TO_DEFAULT_INFO but still doesn't work

Sarahsus commented 2 months ago

For example: if you use srdiff model for training, edit the MODEL_NAME_TO_DEFAULT_INFO: MODEL_NAME_TO_DEFAULT_INFO = { 'IPT_DENOISE': ['ipt', 'configs/ipt/denoise/val_denoise_50.yaml'], 'IPT_DERAIN': ['ipt', 'configs/ipt/derain/val_derain.yaml'], 'IPT_SR_X2': ['ipt', 'configs/ipt/super_resolution/val_sr_x2.yaml'], 'IPT_SR_X3': ['ipt', 'configs/ipt/super_resolution/val_sr_x3.yaml'], 'IPT_SR_X4': ['ipt', 'configs/ipt/super_resolution/val_sr_x4.yaml'], 'FSRCNN_SR_X4': ['fsrcnn', 'configs/fsrcnn/val_sr_x4_Set5_gpu.yaml'], 'CTSDG': ['ctsdg', 'configs/ctsdg/val_inpainting_celeba_gpu.yaml'], 'MIMOUNET': ['mimo_unet', 'configs/mimo_unet/val_deblur_gopro_gpu.yaml'], 'NOAHTCV': ['noahtcv', 'configs/noahtcv/val.yaml'], 'srdiff': ['srdiff', 'configs/srdiff/train.yaml']

}