Method __init__

networks/module.py:351–354

(self, in_channels, base_channels,stage=0)

        return results
class CostRegNet_refine(nn.Module):
    def __init__(self, in_channels, base_channels,stage=0):
        super(CostRegNet_refine, self).__init__()
        self.cosR_small=CostRegNet_part_refine(in_channels, base_channels,stage=0)
        self.cosR_huge=CostRegNet_part_refine(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