scripts.model_conversion.convert_models

scripts.model_conversion.convert_models.convert_duf_model()[source]
scripts.model_conversion.convert_models.convert_edsr(ori_net_path, crt_net_path, save_path, num_block=32)[source]

Convert EDSR models in https://github.com/thstkdgus35/EDSR-PyTorch.

It supports converting x2, x3 and x4 models.

Parameters:

  • ori_net_path (str) – Original network path.

  • crt_net_path (str) – Current network path.

  • save_path (str) – The path to save the converted model.

  • num_block (int) – Number of blocks. Default: 16.

scripts.model_conversion.convert_models.convert_edvr()[source]
scripts.model_conversion.convert_models.convert_esrgan_model()[source]
scripts.model_conversion.convert_models.convert_rcan_model()[source]