c8494f7adf223537db3c6e940cac0eb870ca70a3,models/pointnet2_msg_cls.py,Pointnet2MSG,forward,#Pointnet2MSG#,82

Before Change

    def forward(self, xyz, points=None):
        xyz = xyz.contiguous()
        points = (
            points.transpose(1, 2).contiguous() if points is not None else None
        )
        for module in self.SA_modules:
            xyz, points = module(xyz, points)

        return self.FC_layer(points.squeeze(-1))

After Change

                Each point in the point-cloud MUST
                be formated as (x, y, z, features...)
        
        xyz, features = self._break_up_pc(pointcloud)

        for module in self.SA_modules:
            xyz, features = module(xyz, features)

Instances