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')

参考文献

相关推荐
Python×CATIA工业智造10 分钟前
基于pycatia的CATIA零部件激活状态管理技术解析
python·pycharm·catia二次开发
Julian.zhou26 分钟前
从Manus到OpenManus:AI智能体技术如何重塑未来生活场景?
人工智能·自然语言处理·生活·交互
摸鱼码33 分钟前
(头歌作业—python)3.2 个人所得税计算器(project)
开发语言·python
XTurnV00735 分钟前
Cherry Studio也可以用GPT-4o绘画啦!无需PLUS,无需梯子,国内可用!
人工智能
霖大侠38 分钟前
REVISITING MAE PRE-TRAINING FOR 3D MEDICALIMAGE SEGMENTATION
人工智能·深度学习·机器学习·3d
WHATEVER_LEO39 分钟前
【每日论文】MetaSpatial: Reinforcing 3D Spatial Reasoning in VLMs for the Metaverse
人工智能·深度学习·神经网络·计算机视觉·3d·自然语言处理
狐凄1 小时前
练习题:110
开发语言·python·算法
橙色小博1 小时前
最最最基本神经网络及其原理、程序
人工智能·深度学习·神经网络
大模型真好玩1 小时前
新王登基!DeepSeek-V3-0324横空出世,国产大模型还得看DeepSeek(详细DeepSeek-V3-0324模型评测)
人工智能·深度学习·deepseek
深度学习机器1 小时前
Docling:统一的多文档解析框架 支持多种文档类型的处理与转换
人工智能·github·产品