(self, in_channel, mlp)
| 303 | |
| 304 | class PointNetFeaturePropagation(nn.Module): |
| 305 | def __init__(self, in_channel, mlp): |
| 306 | super(PointNetFeaturePropagation, self).__init__() |
| 307 | self.mlp_convs = nn.ModuleList() |
| 308 | self.mlp_bns = nn.ModuleList() |
| 309 | last_channel = in_channel |
| 310 | for out_channel in mlp: |
| 311 | self.mlp_convs.append(nn.Conv1d(last_channel, out_channel, 1)) |
| 312 | self.mlp_bns.append(nn.BatchNorm1d(out_channel)) |
| 313 | last_channel = out_channel |
| 314 | |
| 315 | def forward(self, xyz1, xyz2, points1, points2): |
| 316 | """ |