2025年torch-fidelity 简便计算FID,ISC,KID,PPL

科技前沿 • 2025-02-20 21:31 • 阅读 56

大家好，我是讯享网，很高兴认识大家。

不要用pytorch-fid，太老了，而且权重文件有问题，你会报EOF.另外不要用pytorch-fid-helper,好像对pytorch部分版本不兼容。

原项目地址:

https://github.com/toshas/torch-fidelity
讯享网https://github.com/toshas/torch-fidelity文档地址:

Usage Examples with Command Line — torch-fidelity 0.3.0 documentationhttps://torch-fidelity.readthedocs.io/en/latest/usage_cmd.html使用方法:

譬如我需要计算两个数据集的FID

想进一步学的的伙伴可以从理论出发，然后自己实现这个fid的计算过程，笔者这里直接搬运了。
原始的fid是一个tensorflow实现 https://github.com/bioinf-jku/TTUR
笔者使用的是pytorch框架，所以使用https://github.com/toshas/torch-fidelity 这个包来计算

pip install torch-fidelity

讯享网

fid计算

其他计算方式看文档或者直接fidelity-help

讯享网fidelity --input1 F:/PYWORK/dataset2/ --input2 F:/PYWORK/dataset1/ -g 0 -f

值得注意的是第一次运行他会下载一个权重文件到文件夹，下的很慢，这里我直接给了，解压完复制到对应文件夹就行。

https://download.csdn.net/download/weixin_/https://download.csdn.net/download/weixin_/除此以外可能会报错

Initializing libiomp5md.dll, but found libiomp5md.dll already initialized.

去你的torch文件夹里把Initializing libiomp5md.dll删了就可以。

2025年torch-fidelity 简便计算FID,ISC,KID,PPL

相关推荐