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

Fix LQGT dataset

Browse Source
This commit is contained in:
James Betker 2020-08-25 11:57:25 -06:00
parent a65b07607c
commit 03eb29a4d9
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
codes/data/LQGT_dataset.py
View File

@ -135,12 +135,12 @@ class LQGTDataset(data.Dataset):
# Enforce force_resize constraints. # Enforce force_resize constraints.
h, w, _ = img_LQ.shape h, w, _ = img_LQ.shape
if h % self.force_multiple != 0 or w % self.force_multiple != 0: if h % self.force_multiple != 0 or w % self.force_multiple != 0:
h, w = (w - w % self.force_multiple), (h - h % self.force_multiple) h, w = (h - h % self.force_multiple), (w - w % self.force_multiple)
img_LQ = cv2.resize(img_LQ, (h, w)) img_LQ = img_LQ[:h, :w, :]
h *= scale h *= scale
w *= scale w *= scale
img_GT = cv2.resize(img_GT, (h, w)) img_GT = img_GT[:h, :w, :]
img_PIX = cv2.resize(img_LQ, (h, w)) img_PIX = img_PIX[:h, :w, :]
if self.opt['phase'] == 'train': if self.opt['phase'] == 'train':
H, W, _ = img_GT.shape H, W, _ = img_GT.shape