월간 데이콘 숫자 3D 이미지 분류 AI 경진대회

알고리즘 | CV | 분류 | 3D | Accuracy

  • moneyIcon 상금 : 100만원 + ɑ
  • 519명 마감

 

Private 3위 (0.97892) PointMLP + Data Augmentation

공동작성자

stroke
2022.09.23 16:27 952 조회 language

안녕하세요~

주어진 데이터, 3D 숫자의 feature를 뽑는 방법으로 point cloud classification에서 높은 성능을 보이는
PointMLP 모델을 사용했습니다.

* PointMLP 모델은  3D pointcloud classification에서 ModelNet40 데이터셋으로 학습된 모델 중 4등을 기록하고 있는 모델입니다

[Rotated data로 Model 실험] (Public score)
Resnet-34: 94.3%
Resnet-50: 95.2%
PointMLP: 96.5%

이상의 결과를 통해 PointMLP 모델을 갖고 가기로 했고 다른 곳에서 추가적인 실험을 진행했습니다.
추가 실험에 대한 설명은 코드 위에 기술하겠습니다.

* VRAM 11GB 사용
** epoch당 10분 소요, 160epoch쯤에서 stop (Data Paste시 epoch당 20분)

코드