mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add grayscale downsampling option

Browse Source
This commit is contained in:
James Betker 2020-05-29 20:34:00 -06:00
parent b123ed8a45
commit 74b313aaa9
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
codes/data/LQGT_dataset.py
View File

@ -5,7 +5,7 @@ import lmdb
import torch
import torch.utils.data as data
import data.util as util
from PIL import Image
from PIL import Image, ImageOps
from io import BytesIO
import torchvision.transforms.functional as F
@ -180,6 +180,9 @@ class LQGTDataset(data.Dataset):
corruption_buffer.seek(0)
img_LQ = Image.open(corruption_buffer)
if self.opt['grayscale']:
img_LQ = ImageOps.grayscale(img_LQ)
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()
img_LQ = F.to_tensor(img_LQ)