Method __init__

networks/module.py:343–346

(self, in_channels, base_channels,stage=0)

class CostRegNet(nn.Module):
    def __init__(self, in_channels, base_channels,stage=0):
        super(CostRegNet, self).__init__()
        self.cosR_small=CostRegNet_part(in_channels, base_channels,stage=0)
        self.cosR_huge=CostRegNet_part(in_channels, base_channels,stage=0)
    def forward(self, x):
        results=torch.cat((self.cosR_small(x),self.cosR_huge(x)),axis=1)
        return results