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

参考文献

相关推荐
im_AMBER1 小时前
学习日志19 python
python·学习
白-胖-子2 小时前
深入剖析大模型在文本生成式 AI 产品架构中的核心地位
人工智能·架构
想要成为计算机高手4 小时前
11. isaacsim4.2教程-Transform 树与Odometry
人工智能·机器人·自动驾驶·ros·rviz·isaac sim·仿真环境
mortimer4 小时前
安装NVIDIA Parakeet时,我遇到的两个Pip“小插曲”
python·github
@昵称不存在4 小时前
Flask input 和datalist结合
后端·python·flask
静心问道4 小时前
InstructBLIP:通过指令微调迈向通用视觉-语言模型
人工智能·多模态·ai技术应用
宇称不守恒4.05 小时前
2025暑期—06神经网络-常见网络2
网络·人工智能·神经网络
赵英英俊5 小时前
Python day25
python
东林牧之5 小时前
Django+celery异步:拿来即用,可移植性高
后端·python·django
何双新5 小时前
基于Tornado的WebSocket实时聊天系统:从零到一构建与解析
python·websocket·tornado