pointnet.pytorch icon indicating copy to clipboard operation
pointnet.pytorch copied to clipboard

Are the MLPs same as in the paper?

Open kuzand opened this issue 2 years ago • 1 comments

In the paper, the first MLP (64, 64) has two layers and the second MLP (64,128,1024) has three layers. In the implementation of PointNetfeat here, the first MLP has just one layer of size 64 and the second MLP has two layers of sizes 128 and 1024 as can be seen from the code below:

self.conv1 = torch.nn.Conv1d(3, 64, 1)
self.conv2 = torch.nn.Conv1d(64, 128, 1)
self.conv3 = torch.nn.Conv1d(128, 1024, 1)

It works fine, but to be consistent with the paper the following modifications can be made:

# First MLP
self.conv1 = torch.nn.Conv1d(3, 64, 1)
self.conv2 = torch.nn.Conv1d(64, 64, 1)

# Second MLP
self.conv3 = torch.nn.Conv1d(64, 64, 1)
self.conv4 = torch.nn.Conv1d(64, 128, 1)
self.conv5 = torch.nn.Conv1d(128, 1024, 1)

and then modify the forward method accordingly.

kuzand avatar Aug 08 '21 11:08 kuzand

I also find this difference, and i noted that there is another issue which have the same question.

chenxl124578 avatar Jun 23 '22 08:06 chenxl124578