| 770 | |
| 771 | class MVSNet(nn.Module): |
| 772 | def __init__(self, |
| 773 | num_groups=1, |
| 774 | norm_act=InPlaceABN, |
| 775 | levels=1): |
| 776 | super(MVSNet, self).__init__() |
| 777 | self.levels = levels # 3 depth levels |
| 778 | self.n_depths = [128,32,8] |
| 779 | self.G = num_groups # number of groups in groupwise correlation |
| 780 | self.feature = FeatureNet() |
| 781 | |
| 782 | self.N_importance = 0 |
| 783 | self.chunk = 1024 |
| 784 | |
| 785 | self.cost_reg_2 = CostRegNet(32+9, norm_act) |
| 786 | |
| 787 | def build_volume_costvar(self, feats, proj_mats, depth_values, pad=0): |
| 788 | # feats: (B, V, C, H, W) |