lpips使用笔记

LPIPS指的是 Learned Perceptual Image Patch Similarity(学习感知图像块相似度),它是一种用于衡量两张图像之间感知相似度的指标。

1. 安装

bash 复制代码
pip install lpips

2. 使用

python 复制代码
import torch
import lpips

# 加载预训练的 LPIPS 模型
loss_fn = lpips.LPIPS(net='vgg')

# 假设您有两个图像张量 img1 和 img2
img1 = torch.rand(1, 3, 256, 256)  # 示例随机图像张量 1
img2 = torch.rand(1, 3, 256, 256)  # 示例随机图像张量 2

# 计算 LPIPS 距离
d = loss_fn(img1, img2)
print(d)
python 复制代码
import torch
import lpips

# 假设您已下载的模型路径为'model_path'
loss_fn = lpips.LPIPS(net_type='vgg', model_path='model_path')

参考文献

相关推荐
萧鼎12 分钟前
RAGFlow:构建高效检索增强生成流程的技术解析
人工智能·python
爱的叹息16 分钟前
主流开源 LLM 应用开发平台详解
人工智能·开源
赋范大模型技术社区18 分钟前
从0手撕代码搭建MCP Client与Server!详解DeepSeek、ollama、vLLM接入MCP实战!
人工智能·mcp
cooljser22 分钟前
告别手动操作!用脚本搞定小程序签到的全过程
python
Baihai_IDP28 分钟前
面对开源大模型浪潮,基础模型公司如何持续盈利?
人工智能·openai·deepseek
陈明勇29 分钟前
MCP 实战:用 Go 语言开发一个查询 IP 信息的 MCP 服务器
人工智能·后端·mcp
浏览器爱好者35 分钟前
如何下载适用于语音识别功能增强的Google Chrome浏览器
人工智能·chrome·语音识别
凌叁儿38 分钟前
从零开始搭建Django博客①--正式开始前的准备工作
python·django·sqlite
孔令飞1 小时前
彻底学会 gRPC:用 Go 实现一个迷你考试服务
人工智能·云原生·go
梓羽玩Python1 小时前
告别OCR!这个AI文档神器直接"看懂"PDF,支持文档归类及多模态问答!
人工智能·github