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)