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

参考文献

相关推荐
言德斐1 天前
Python Web框架深度对比:Django vs Flask vs FastAPI(含优缺点与选型策略)
前端·python·django
TMT星球1 天前
加速进化发布Booster K1,打造AI时代的苹果公司
人工智能
开心-开心急了1 天前
Flask入门教程——李辉 第5章: 数据库 关键知识梳理
笔记·后端·python·flask·1024程序员节
tangchen。1 天前
YOLOv4 :兼顾速度与精度!
人工智能·计算机视觉·目标跟踪
郑清1 天前
Spring AI Alibaba 10分钟快速入门
java·人工智能·后端·ai·1024程序员节·springaialibaba
学术头条1 天前
用视觉压缩文本!清华、智谱推出Glyph框架:通过视觉-文本压缩扩展上下文窗口
人工智能·深度学习·计算机视觉
Mrliu__1 天前
Opencv(一): 用Opencv了解图像
人工智能·opencv·计算机视觉
CodeCraft Studio1 天前
国产化Excel开发组件Spire.XLS教程:在Python中将Pandas DataFrame导出到Excel的详细教程
python·excel·pandas
墨香幽梦客1 天前
掌控制造脉络:电子元器件行业常用ERP系统概览与参考指南
大数据·人工智能
B站_计算机毕业设计之家1 天前
python舆情分析可视化系统 情感分析 微博 爬虫 scrapy爬虫技术 朴素贝叶斯分类算法大数据 计算机✅
大数据·爬虫·python·scrapy·数据分析·1024程序员节·舆情分析