From 1eb9c5a059a6cfaa3c826479f38b6bbe536d24a2 Mon Sep 17 00:00:00 2001 From: James Betker Date: Fri, 29 May 2020 22:04:50 -0600 Subject: [PATCH] Fix grayscale --- codes/data/LQGT_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/data/LQGT_dataset.py b/codes/data/LQGT_dataset.py index b8321963..47a2859d 100644 --- a/codes/data/LQGT_dataset.py +++ b/codes/data/LQGT_dataset.py @@ -181,7 +181,7 @@ class LQGTDataset(data.Dataset): img_LQ = Image.open(corruption_buffer) if self.opt['grayscale']: - img_LQ = ImageOps.grayscale(img_LQ) + img_LQ = ImageOps.grayscale(img_LQ).convert('RGB') img_GT = torch.from_numpy(np.ascontiguousarray(np.transpose(img_GT, (2, 0, 1)))).float() img_PIX = torch.from_numpy(np.ascontiguousarray(np.transpose(img_PIX, (2, 0, 1)))).float()