def encode_real_B(self):
if self.opt.whether_encode_cloth and self.opt.which_image_encode == 'groundTruth':
clip_start_index = (self.opt.fineSize-self.opt.encode_size)//2
clip_end_index = clip_start_index + self.opt.encode_size
clip_cloth = self.input_B[:,:,clip_start_index:clip_end_index,clip_start_index:clip_end_index ]
self.z_encoded = self.encode(clip_cloth)
elif self.opt.which_image_encode == 'contour':
self.z_encoded = self.encode(self.input_A)
else:
self.z_encoded = self.encode(self.input_B)
return util.tensor2vec(self.z_encoded)
def encode_real_B(self):
if self.opt.whether_encode_cloth and self.opt.which_image_encode == 'groundTruth':
#real_A=util.tensor2im(Variable(self.input_A).data)
#from skimage import io
#io.imshow(real_A)
clip_start_index = (self.opt.fineSize-self.opt.encode_size)//2
clip_end_index = clip_start_index + self.opt.encode_size
clip_cloth = self.input_B[:,:,clip_start_index:clip_end_index,clip_start_index:clip_end_index ]
self.z_encoded = self.encode(clip_cloth)
elif self.opt.which_image_encode == 'contour':
self.z_encoded = self.encode(self.input_A)
else:
self.z_encoded = self.encode(self.input_B)
return util.tensor2vec(self.z_encoded)
def encode_real_B(self):
self.z_encoded = self.encode(self.input_B)
return util.tensor2vec(self.z_encoded)
def encode_real_B(self):
# self.z_encoded = self.encode(self.input_B)
self.z_encoded, _ = self.netE.forward(self.input_B)
return util.tensor2vec(self.z_encoded)
