def colorize(self, img): img, imgResizeParams = pad_image(img, 256) dataset = self.__createSingleImageDataset(img) self.model.set_input(dataset[0]) self.model.test() visuals = self.model.get_current_visuals() result_img_array = tensor2im(visuals['fake_B']) result_img = Image.fromarray(result_img_array) return unpad_image(result_img, imgResizeParams)
def transferStyle(self, skeletonImg, styleImg): skeletonImg, imgResizeParams = pad_image(skeletonImg, (2048, 256)) styleImg, _ = pad_image(styleImg, (2048, 256)) dataset = self.__createSingleImageDataset(skeletonImg, styleImg) self.model.set_input(dataset[0]) self.model.test() visuals = self.model.get_current_visuals() result_img_array = tensor2im(visuals['fake_B']) result_img = Image.fromarray(result_img_array) return unpad_image(result_img, imgResizeParams)