쑤야의 개발 블로그

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

먼저 PointNetCls 클래스의 전체 코드를 살펴보며 PointNetCls가 Figure 상에서 어떤 부분을 담당하는지 파악해 보았다.  클래스 전체 코드는 아래와 같다. class PointNetCls(nn.Module): def __init__(self, k=2, feature_transform=False): super(PointNetCls, self).__init__() self.feature_transform = feature_transform self.feat = PointNetfeat(global_feat=True, feature_transform=feature_transform) self.fc1 = nn.Linear(1024, 512..

AI/PyTorch 2024. 8. 24. 14:58

파이토치를 공부할 겸 포인트 클라우드 딥러닝 모델 중 유명한 PointNet 코드를 분석해 보도록 하겠다.  공부용으로 선택한 코드는 papers with code라는 사이트에서 PointNet을 검색하여 파이토치로 구현된 코드들 중 가장 많은 스타를 받은 코드로 선택했다.  papers with code의 경우 깃허브에 올라온(?) 딥러닝 모델 코드들을 모아놓은 사이트라 AI 관련해서 공부하는 사람들에게 굉장히 유용한 사이트인 것 같다.  https://paperswithcode.com/paper/pointnet-deep-learning-on-point-sets-for-3d#code Papers with Code - PointNet: Deep Learning on Point Sets for 3D Cla..

AI/PyTorch 2024. 8. 24. 14:36