dvlab-research/PointGroup

[help wanted] A question in model implementation

JLUtangchuan opened this issue · 2 comments

JLUtangchuan commented

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 commented

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

JLUtangchuan commented

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

  • 👍1