def _get_patch(self, hr, filename):
patch_size = self.args.patch_size
if self.train:
scale = self.scale[0]
if self.args.task_type == 'denoising':
lr, hr = common.get_patch_noise(
hr, patch_size, scale
)
if self.args.task_type == 'SISR':
lr, hr = common.get_patch_bic(
hr, patch_size, scale
)
if self.args.task_type == 'JIAR':
lr, hr = common.get_patch_compress(
hr, patch_size, scale
)
lr, hr = common.augment([lr, hr])
return lr, hr, scale
else:
scale = self.scale[0]
if self.args.task_type == 'denoising':
lr, hr = common.add_img_noise(
hr, scale
)
if self.args.task_type == 'SISR':
lr, hr = self._get_patch_test(
hr, patch_size, scale
)
if self.args.task_type == 'JIAR':
lr, hr = common.get_img_compress(
hr, patch_size, scale
)
return lr, hr, scale
def _get_patch(self, hr, lr):
import pdb; pdb.set_trace()
patch_size = self.args.patch_size
#print(hr.shape)
if self.train:
scale = self.scale[0]
if self.args.task_type == 'denoising':
lr, hr = lr, hr #common.get_patch_noise( hr, patch_size, scale)
if self.args.task_type == 'SISR':
lr, hr = common.get_patch_bic( hr, patch_size, scale)
if self.args.task_type == 'JIAR':
lr, hr = common.get_patch_compress( hr, patch_size, scale)
#lr, hr = common.augment([lr, hr])
return lr, hr, scale
else:
scale = self.scale[0]
if self.args.task_type == 'denoising':
lr, hr = lr, hr #common.get_img_noise( hr, patch_size, scale)
if self.args.task_type == 'SISR':
lr, hr = self._get_patch_test( hr, patch_size, scale)
if self.args.task_type == 'JIAR':
lr, hr = common.get_img_compress( hr, patch_size, scale)
return lr, hr, scale
def _get_patch(self, hr, filename):
patch_size = self.args.patch_size
if self.train:
scale = self.scale[0]
#print("GETTING PATCH", hr.shape)
if self.args.task_type == 'denoising':
lr, hr = common.get_patch_noise(hr, patch_size, scale)
if self.args.task_type == 'SISR':
lr, hr = common.get_patch_bic(hr, patch_size, scale)
if self.args.task_type == 'JIAR':
lr, hr = common.get_patch_compress(hr, patch_size, scale)
if self.args.task_type == 'clip':
lr, hr = common.get_patch_clip(hr, self.args.ev, 0, 255)
if self.args.task_type == 'log':
lr, hr = common.linearize_and_scale(hr,
scale_to=2.0,
clip_at=1.0)
#print("LR", lr[0, :5],"\nHR", hr[0, :5])
lr, hr = common.augment([lr, hr])
return lr, hr, scale
else:
scale = self.scale[0]
if self.args.task_type == 'denoising':
lr, hr = common.get_img_noise(hr, patch_size, scale)
if self.args.task_type == 'SISR':
lr, hr = self._get_patch_test(hr, patch_size, scale)
if self.args.task_type == 'JIAR':
lr, hr = common.get_img_compress(hr, patch_size, scale)
if self.args.task_type == 'clip':
lr, hr = common.get_patch_clip(hr, self.args.ev, 0, 255)
if self.args.task_type == 'log':
lr, hr = common.linearize_and_scale(hr,
scale_to=2.0,
clip_at=1.0)
return lr, hr, scale