def __init__(self,const): super(ConcatSVM,self).__init__() self.const = copy.deepcopy(const) self.mlp = pytorch_layers.create_mlp(self.const.mlp_const) self.bn = nn.BatchNorm1d(self.const.mlp_const['in_dim']) print('2D-SVM') self.w = nn.Parameter(torch.FloatTensor([0,0,0]))
def __init__(self, const): super(ConcatSVM, self).__init__() self.const = copy.deepcopy(const) self.mlp = pytorch_layers.create_mlp(self.const.mlp_const) self.bn = nn.BatchNorm1d(6)
def __init__(self, const): super(VerbGivenBoxesAndObjectLabel, self).__init__() self.const = copy.deepcopy(const) self.mlp = pytorch_layers.create_mlp(self.const.mlp_const)
def __init__(self,const): super(VerbGivenObjectAppearance,self).__init__() self.const = copy.deepcopy(const) self.mlp = pytorch_layers.create_mlp(self.const.mlp_const)
def __init__(self, const): super(VerbGivenHumanPose, self).__init__() self.const = copy.deepcopy(const) self.mlp = pytorch_layers.create_mlp(self.const.mlp_const)