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