【图像生成模型评价指标】-- HPS及实践

原创

已于 2023-11-06 22:26:11 修改 · 1k 阅读

标签

#人工智能 #stable diffusion #AIGC #prompt

收录于

于 2023-11-06 22:24:22 首次发布

文章介绍了如何使用HPSV2模型对生成的图片进行打分测试，通过提供一组图片和提示，结果显示模型认为与提示最匹配的图片顺序。

【图像生成模型评价指标】-- HPS及实践

HPS V2 项目地址
 测试数据下载地址

一、用HPS V2给模型生成的图片打分测试

Step1: 在本地通过git和pip安装HPS V2

git clone https://github.com/tgxs002/HPSv2.git
cd HPSv2 && python -m pip install .

Step2: 加载模型和图像进行测试(我这里抽取了test中的前八张，用对应的prompt进行了测试)

import hpsv2

imgs_path = ["/content/drive/MyDrive/ColabNotebooks/GenerateModelMetrix/HPSv2/hpsv2/tests/my_test/test/00000.jpg",
"/content/drive/MyDrive/ColabNotebooks/GenerateModelMetrix/HPSv2/hpsv2/tests/my_test/test/00001.jpg",
"/content/drive/MyDrive/ColabNotebooks/GenerateModelMetrix/HPSv2/hpsv2/tests/my_test/test/00002.jpg",
"/content/drive/MyDrive/ColabNotebooks/GenerateModelMetrix/HPSv2/hpsv2/tests/my_test/test/00003.jpg",
"/content/drive/MyDrive/ColabNotebooks/Generate