lpips_perceptual_distance_alexnet_quickstart.py

1import lpips
2import torch
3
4loss_fn_alex = lpips.LPIPS(net='alex') # best forward scores
5
6img0 = torch.zeros(1,3,64,64) # image should be RGB, normalized to [-1,1]
7img1 = torch.zeros(1,3,64,64)
8
9d = loss_fn_alex(img0, img1)
10
11print(f"Distance: {d.item()}")