torch-fidelity 简便计算FID,ISC,KID,PPL
不要用pytorch-fid,太老了,而且权重文件有问题,你会报EOF.另外不要用pytorch-fid-helper,好像对pytorch部分版本不兼容。原项目地址:https://github.com/toshas/torch-fidelityhttps://github.com/toshas/torch-fidelity文档地址:Usage Examples with Command Lin
不要用pytorch-fid,太老了,而且权重文件有问题,你会报EOF.另外不要用pytorch-fid-helper,好像对pytorch部分版本不兼容。
原项目地址:
https://github.com/toshas/torch-fidelityhttps://github.com/toshas/torch-fidelity文档地址:
譬如我需要计算两个数据集的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
值得注意的是第一次运行他会下载一个权重文件到文件夹,下的很慢,这里我直接给了,解压完复制到对应文件夹就行。
Initializing libiomp5md.dll, but found libiomp5md.dll already initialized.
去你的torch文件夹里把Initializing libiomp5md.dll删了就可以。
开放原子开发者工作坊旨在鼓励更多人参与开源活动,与志同道合的开发者们相互交流开发经验、分享开发心得、获取前沿技术趋势。工作坊有多种形式的开发者活动,如meetup、训练营等,主打技术交流,干货满满,真诚地邀请各位开发者共同参与!
更多推荐
所有评论(0)