Spaces:
Runtime error
Runtime error
import torch | |
from torch import nn | |
class BaseColor(nn.Module): | |
def __init__(self): | |
super(BaseColor, self).__init__() | |
self.l_cent = 50. | |
self.l_norm = 100. | |
self.ab_norm = 110. | |
def normalize_l(self, in_l): | |
return (in_l-self.l_cent)/self.l_norm | |
def unnormalize_l(self, in_l): | |
return in_l*self.l_norm + self.l_cent | |
def normalize_ab(self, in_ab): | |
return in_ab/self.ab_norm | |
def unnormalize_ab(self, in_ab): | |
return in_ab*self.ab_norm | |