File size: 430 Bytes
81f9834 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 | from utils.flolpips import Flolpips
import torch
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
eval_metric = Flolpips().to(device)
batch = 8
I0 = torch.rand(8, 3, 256, 448).to(device)
I1 = torch.rand(8, 3, 256, 448).to(device)
frame_dis = torch.rand(8, 3, 256, 448).to(device)
frame_ref = torch.rand(8, 3, 256, 448).to(device)
flolpips = eval_metric.forward(I0, I1, frame_dis, frame_ref)
print(flolpips) |