关于FID的计算,这里我直接使用startGANv2中的计算FID的方法,比较简单,快速。理论部分我就不讲了,网上一搜一大堆,我写的都是干货(主要是怕太长了,没人看😂)
上一篇是关于PSNR、SSIM、LPIPS的感兴趣的小伙伴们可以看一下。
https://blog.csdn.net/qq_43826289/article/details/135396128?spm=1001.2014.3001.5502
1、首先进入你所在项目的文件夹下(或者你所测试的数据集同级目录下)下载下面代码到你的项目中
https://gitcode.com/clovaai/stargan-v2/overview?utm_source=csdn_github_accelerator&isLogin=1
如下图所示
2、需要根据上面给出的地址配置所需要的虚拟环境,不用按照给出的版本,根据自己的环境版本安装对应的版本即可(一般来说,直接安装最新版本就可以,我也没管版本是多少)
比如说原论文中安装的版本
pip install pillow==7.0.0 scipy==1.2.1 tqdm==4.43.0 munch==2.5.0
换成下面的版本也可以(下载的是最新版本的)
pip install pillow scipy tqdm munch
3、配置完虚拟环境之后
配置完环境将激活虚拟环境,进入下载代码的文件夹
cd stargan-v2/
然后执行下面的命令
python -m metrics.fid --paths 原始文件 输出文件
例如,可以是单张图像,也可以是一个文件夹,这里我使用的是一个文件夹
python -m metrics.fid --path ../data/input ../data/output
然后等待程序运行完就可以得到FID。
用于日常记录,希望能帮助到大家!!!!!最后祝大家都科研顺利,多多发论文!!!!!