[help wanted] A question in model implementation

[JLUtangchuan]

See in pointgroup.py#L283, why semantic_preds >1 ? (ignore Class 0 and Class 1?)

object_idxs = torch.nonzero(semantic_preds > 1).view(-1)

PointGroup/model/pointgroup/pointgroup.py

Line 283 in dea7ab1

object_idxs = torch.nonzero(semantic_preds > 1).view(-1)

[QingWindIsStillTheWind]

Do you know why this is set? I also have this doubt

[JLUtangchuan]

class0 and class1 represent wall and floor, those should be ignored during offset prediction