Transpose for part segmentation

[EdwardSmith1884]

Hi,
I have a quick question about the different outputs of pointnet and DGCNN for segmentation. It looks like for pointnet you transpose the output of the network (batchsize, number of parts, number of points) -> (batchsize, number of points, number of parts) :

vnn/models/pointnet_partseg.py

Line 70 in 1300091

net = net.transpose(2, 1).contiguous()

But for DGCNN this does not happen despite the output of the network also being (batchsize, number of parts, number of points):

vnn/models/dgcnn_partseg.py

Line 108 in 1300091

return x, trans_feat

Why is this the case? Is there then not a mismatch with ground truth segmentation labels for one of the networks?

Thanks for your help!

[FlyingGiraffe]

Hi,

Thanks for the comment! This is indeed a bug -- I merged the DGCNN models into the PointNet training framework but didn't check all the details. Already updated! Also, I've created this repo with separate PointNet and DGCNN training setups, which may be of some help. Thanks again for noticing this bug!

Best,
Congyue

[EdwardSmith1884]

Great! thanks for the update!