Open Simon-ssp opened 3 months ago
I added "basicvsr" to MODEL_NAME_TO_DEFAULT_INFO but still doesn't work
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']
}
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'