Skip to content

make_custom_stats overcount files in directory #68

@WeizhenWang-1210

Description

@WeizhenWang-1210

Suppose you have a folder named with path "D:/test", containing only one image "000391.png".

 -D:/
    -test:/
        -000391.png

The following script

from cleanfid import fid
if __name__ == "__main__":
    fid.make_custom_stats("test", "D:/test", mode="clean")

will have this output

(srkl2) PS D:\SRK-Diffusion> python -m clean-fid
Found 2 images in the folder D:/test
custom stats: test : 100%|███████████████████████████████████████████████████████████████| 1/1 [00:00<00:00,  2.48it/s]
saving custom FID stats to F:\Anaconda\envs\srkl2\lib\site-packages\cleanfid\stats\test_clean_custom_na.npz
saving custom KID stats to F:\Anaconda\envs\srkl2\lib\site-packages\cleanfid\stats\test_clean_custom_na_kid.npz

Which is inconsistent with the number of files contained under D:/test

I am running on a windows11 machine with RTX 4090 GPU.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions