def __init__(self, net, dev, lids, w, image, k, s, sem_style, sem_cont,
gamma, gamma_scale):
super(SemanticStyle.Loss, self).__init__(net, dev, lids, w, image,
k, s)
self.sem_style = to_torch(sem_style).to(dev)
self.sem_cont = to_torch(sem_cont).to(dev)
self.gamma = gamma
self.gamma_scale = gamma_scale
def _get_or_create_seed(self, content, style, x):
if x is None:
if content.image is None:
x = image.new_random_white((256, 256, 3), mean=style.image)
else:
x = image.new_random_white(content.image.shape,
mean=content.image)
return image.to_torch(x).to(self.backbone.dev).requires_grad_()
def __init__(self, net, dev, lids, w, image):
self.lids = lids
self.w = w
self.net = net
self.image = to_torch(image).to(dev)
def __init__(self, net, dev, lid, image):
self.net = net
self.lid = lid
self.dev = dev
self.image = to_torch(image).to(dev)