Fix image color.

test.py

@@ -24,6 +24,7 @@ base = os.path.splitext(os.path.basename(img_path))[0]
 # read image
 print(img_path);
 img = cv2.imread(img_path, cv2.IMREAD_GRAYSCALE)
+img = img * 1.0 / 255
 img = torch.from_numpy(img[np.newaxis, :, :]).float()
 img_LR = img.unsqueeze(0)
 img_LR = img_LR.to(device)
@@ -31,5 +32,7 @@ img_LR = img_LR.to(device)
 print('Start upscaling...')
 output = model(img_LR).squeeze(dim=0).float().cpu().clamp_(0, 1).numpy()
 output = np.transpose(output, (1, 2, 0))
+output = (output * 255.0).round()
 print('Finished upscaling, saving image.')
+print(output_dir)
 cv2.imwrite(output_dir, output, [int(cv2.IMWRITE_JPEG_QUALITY), 90])