lpips_perceptual_image_distance_with_vgg_network.py

1import lpips
2import torch
3
4loss_fn_vgg = lpips.LPIPS(net='vgg') # can also use net='alex' or 'squeeze'
5
6img0 = torch.zeros(1,3,64,64) # image should be RGB, normalized to [-1,1]
7img1 = torch.randn(1,3,64,64)
8
9d = loss_fn_vgg(img0, img1)
10
11print(f'Distance: {d.item()}')