(img1, img2, value_range=(0, 1))
| 71 | |
| 72 | loss_fn_vgg = None |
| 73 | def lpips(img1, img2, value_range=(0, 1)): |
| 74 | global loss_fn_vgg |
| 75 | if loss_fn_vgg is None: |
| 76 | loss_fn_vgg = LPIPS(net='vgg').cuda().eval() |
| 77 | # normalize to [-1, 1] |
| 78 | img1 = (img1 - value_range[0]) / (value_range[1] - value_range[0]) * 2 - 1 |
| 79 | img2 = (img2 - value_range[0]) / (value_range[1] - value_range[0]) * 2 - 1 |
| 80 | return loss_fn_vgg(img1, img2).mean() |
| 81 | |
| 82 | |
| 83 | def normal_angle(pred, gt): |
no test coverage detected