2313fbdad72507c0af7daaf43d1887fc95453762,src/pytorch_fid/fid_score.py,,_compute_statistics_of_path,#Any#Any#Any#Any#Any#,220

Before Change

        f.close()
    else:
        path = pathlib.Path(path)
        files = list(path.glob("*.jpg")) + list(path.glob("*.png"))
        m, s = calculate_activation_statistics(files, model, batch_size,
                                               dims, device)

After Change

        f.close()
    else:
        path = pathlib.Path(path)
        files = [file for ext in IMAGE_EXTENSIONS
                 for file in path.glob("*.{}".format(ext))]
        m, s = calculate_activation_statistics(files, model, batch_size,
                                               dims, device)

    return m, s

Instances