)
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:
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)