如何计算fid、kid和is指标（附代码）

废话不多说，直接上代码：

import torch_fidelity

#  fid、kid和is
metrics_dict = torch_fidelity.calculate_metrics(
    input1='fake',
    input2='real',
    cuda=True, 
    isc=True, 
    fid=True, 
    kid=True, 
    prc=True, 
    verbose=False,
    kid_subset_size=62
)

import yaml

def save_dict_to_yaml(dict_value: dict, save_path: str):
    """dict保存为yaml"""
    with open(save_path, 'w') as file:
        file.write(yaml.dump(dict_value, allow_unicode=True))


def read_yaml_to_dict(yaml_path: str, ):
    with open(yaml_path) as file:
        dict_value = yaml.load(file.read(), Loader=yaml.FullLoader)
        return dict_value




save_dict_to_yaml(metrics_dict, 'result.yaml')