计算压缩前后两张jpg格式的灰度图像的DISTS和LPIPS

要计算压缩前后两张jpg格式的灰度图像的DISTS和LPIPS，你需要先安装相应的Python库。DISTS和LPIPS是通过计算两张图像的特征向量之间的距离来衡量图像之间的差异，因此你需要使用预训练好的模型来提取特征向量。以下是一个简单的示例代码，可以帮助你计算DISTS和LPIPS：

import torch
import torchvision
import cv2
from skimage import img_as_float
from lpips import LPIPS
from DISTS import DISTS

# 加载预训练模型
lpips_model = LPIPS(net='alex', verbose=False)
dists_model = DISTS()

# 读取原始图像和压缩后的图像
img_original = img_as_float(cv2.imread('original.jpg', cv2.IMREAD_GRAYSCALE))
img_compressed = img_as_float(cv2.imread('compressed.jpg', cv2.IMREAD_GRAYSCALE))

# 将图像转换为PyTorch张量
img_original_tensor = torchvision.transforms.functional.to_tensor(img_original)
img_compressed_tensor = torchvision.transforms.functional.to_tensor(img_compressed)

# 提取特征向量
feat_original = lpips_model.forward(img_original_tensor.unsqueeze(0))
feat_compressed = lpips_model.forward(img_compressed_tensor.unsqueeze(0))

# 计算LPIPS
lpips_distance = torch.nn.functional.pairwise_distance(feat_original, feat_compressed).item()
print('LPIPS distance:', lpips_distance)

# 计算DISTS
dists_distance = dists_model.forward(img_original, img_compressed)
print('DISTS distance:', dists_distance)

这个代码使用了`torchvision`、`cv2`、`skimage`等Python库，需要提前安装。其中，`lpips`和`DISTS`分别是计算LPIPS和DISTS距离的Python库，需要提前安装。在计算LPIPS和DISTS距离之前，我们需要先将图像转换为PyTorch张量，并使用预训练模型提取特征向量。最后，我们分别计算LPIPS和DISTS距离，并输出结果。

需要注意的是，DISTS和LPIPS都是通过计算特征向量之间的距离来衡量图像之间的差异，因此它们的取值范围与SSIM不同。在使用这些指标时，需要根据具体情况来选择合适的阈值。