图像评价指标PSNR、SSIM、LPIPS、FID(接上篇,这篇主要讲FID)

2024-01-08 16:55:00

关于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。
在这里插入图片描述用于日常记录,希望能帮助到大家!!!!!最后祝大家都科研顺利,多多发论文!!!!!

文章来源:https://blog.csdn.net/qq_43826289/article/details/135396468
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。