PointNet 코드 분석 - PointNetfeat
PointNetfeat이 논문의 Figure에서 지칭하는 부분을 미리 알아봤었다. 아래 그림의 빨간 영역이다. 클래스 전체 코드는 아래와 같다. class PointNetfeat(nn.Module): def __init__(self, global_feat = True, feature_transform = False): super(PointNetfeat, self).__init__() self.stn = STN3d() self.conv1 = torch.nn.Conv1d(3, 64, 1) self.conv2 = torch.nn.Conv1d(64, 128, 1) self.conv3 = torch.nn.Conv1d(128, 1024, 1)..
AI/PyTorch
2024. 8. 24. 15:25