def _get_patch(self, img_in, img_tar,img_in4,img_in2):
if self.train:
if self.fullTrain:
scale = self.scale[self.idx_scale]
img_in, img_tar,img_in4,img_in2, pi = common_DW.get_patch(
img_in, img_tar,img_in4,img_in2, self.args, scale)
img_in, img_tar,img_in4,img_in2, ai = common_DW.augment(img_in, img_tar,img_in4,img_in2)
return img_in, img_tar,img_in4,img_in2, pi, ai
else:
scale = int(self.fullInputScale[0])/int(self.fullTargetScale[0])
img_in, img_tar,img_in4,img_in2, pi = common_DW.get_patch_DW(
img_in, img_tar, self.args, int(scale))
img_in, img_tar,img_in4,img_in2, ai = common_DW.augment_DW(img_in, img_tar)
return img_in, img_tar,None,None, pi, ai
else:
scale = int(self.fullInputScale[0])/int(self.fullTargetScale[0]) #self.scale[self.idx_scale] ##0209 revised.
#print(img_tar.shape)
#print(img_in.shape)
scale = 1
ih, iw, c = img_in.shape
img_tar = img_tar[30:ih *scale-30, 30:iw * scale-30, :]
img_in = img_in[30:ih * scale-30, 30:iw * scale-30, :]
#print(img_in.shape)
#print(img_tar.shape)
return img_in, img_tar, None, None,None,None
def _get_patch(self, img_in, img_tar, img_in4, img_in2):
if self.train:
if self.fullTrain:
scale = self.scale[self.idx_scale]
img_in, img_tar, img_in4, img_in2, pi = common_DW.get_patch(
img_in, img_tar, img_in4, img_in2, self.args, scale)
img_in, img_tar, img_in4, img_in2, ai = common_DW.augment(
img_in, img_tar, img_in4, img_in2)
return img_in, img_tar, img_in4, img_in2, pi, ai
else:
scale = int(self.fullInputScale[0]) / int(
self.fullTargetScale[0])
img_in, img_tar, img_in4, img_in2, pi = common_DW.get_patch_DW(
img_in, img_tar, self.args, int(scale))
img_in, img_tar, img_in4, img_in2, ai = common_DW.augment_DW(
img_in, img_tar)
return img_in, img_tar, None, None, pi, ai
else:
scale = 2 #%self.scale[self.idx_scale] ##0209 revised.
ih, iw, c = img_in.shape
print(img_in.shape)
print(img_tar.shape)
# img_in_2 = img_in[(ih//2)-50:(ih//2)+50,(iw//2)-50,(iw//2)+50,:]
# img_tar_2 = img_tar[(ih//2)*2-100: (ih//2)*2+100,(iw//2)*2-100:(iw//2)*2:-100,:]
# img_in_2 = img_in[20:ih-20,20:iw-20,:]
# img_tar_2 = img_tar[40:ih*2-40,40:iw*2-40,:]
img_in, img_tar, img_in4, img_in2, pi = common_DW.get_patch_DW_test(
img_in, img_tar, self.args, int(scale))
img_in, img_tar, img_in4, img_in2, ai = common_DW.augment_DW(
img_in, img_tar)
print(img_in.shape)
print(img_tar.shape)
#img_tar = img_tar[0:ih * scale, 0:iw * scale, :]
return img_in, img_tar, None, None, None, None