|
|
|
@ -52,8 +52,8 @@ lut_model = LoadCheckpoint(config.lut_model_path).cuda()
|
|
|
|
|
print(net_model)
|
|
|
|
|
print(lut_model)
|
|
|
|
|
|
|
|
|
|
lr_image = cv2.imread(str(config.project_path / "data" / "Set14/LR/X4/monarch.png"))[:,::-1,::-1].copy()
|
|
|
|
|
image_gt = cv2.imread(str(config.project_path / "data" / "Set14/HR/monarch.png"))[:,::-1,::-1].copy()
|
|
|
|
|
lr_image = cv2.imread(str(config.lr_image_path))[:,:,::-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()
|
|
|
|
|
|
|
|
|
|