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
时间: 2024-02-01 09:12:08 浏览: 125
lbcnn.torch-master.zip_..累lbcnn;x3_LBCNN_lbp_lbp 神经网络_torch
这是一个定义了BaseColor类的代码片段,该类继承自nn.Module。在初始化函数__init__()中,定义了三个成员变量:l_cent、l_norm、ab_norm。normalize_l()和unnormalize_l()分别是将输入的亮度值进行归一化和反归一化的函数;normalize_ab()和unnormalize_ab()分别是将输入的色度值进行归一化和反归一化的函数。这些函数都是为了方便对图像进行处理而设计的。
阅读全文