这是一个用飞桨计算弗雷歇 Inception 距离(FID)的仓库,迁移它的 PyTorch 实现和 PaddleGAN 中的实现。
先克隆本仓库,再使用 pip 安装:
git clone https://github.com/BrilliantYuKaimin/paddle-fid
pip install -e paddle-fid
运行下面的命令可以计算两个目录中的图片间的 FID:
python -m paddle_fid 图片目录甲 图片目录乙
如果安装了 GPU 版本的飞桨却想使用 CPU 进行推理,可以添加选项 --device cpu
。