main
Vladimir Protsenko 7 months ago
parent 96868011bc
commit 0597cfba9f

@ -52,8 +52,8 @@ lut_model = LoadCheckpoint(config.lut_model_path).cuda()
print(net_model) print(net_model)
print(lut_model) print(lut_model)
lr_image = cv2.imread(str(config.project_path / "data" / "Set14/LR/X4/monarch.png"))[:,::-1,::-1].copy() lr_image = cv2.imread(str(config.lr_image_path))[:,:,::-1].copy()
image_gt = cv2.imread(str(config.project_path / "data" / "Set14/HR/monarch.png"))[:,::-1,::-1].copy() image_gt = cv2.imread(str(config.hr_image_path))[:,:,::-1].copy()
input_image = torch.tensor(lr_image).type(torch.float32).permute(2,0,1)[None,...].cuda() input_image = torch.tensor(lr_image).type(torch.float32).permute(2,0,1)[None,...].cuda()

Loading…
Cancel
Save