(2021.03.10~2021.04.14)
Dacon 모션 키포인트 검출 AI경진대회는 사람이 어떤 운동을 수행할 때, 각각의 동작마다 신체 부위에서 keypoint를 검출하는 대회이다.
1.open
├── data
│ ├── test_imgs.zip
│ ├── train_imgs.zip
│ ├── test_imgs
│ └── train_imgs
├── train_df.csv
└── sample_submission대회측에서 제공하는 데이터는 train 이미지 4195장,test 이미지 1600장, train이미지의 이름과 keypoint 24 지점의 x,y 좌표가 적혀있는 .csv파일,정답지로 구성되어있다.
- EDA : dacon에서 제공하는 이미지와 keypoint지점을 사진위에 표현하며 제공된 데이터가 정확한 좌표를 포함하는지 확인
-
Preprocessing : keypoint matching이 제대로 되지 않은 데이터 총 101개를 제거
-
Augmentation : albumentation library를 이용. HorizontalFlip, RandomRotate90, MotionBlur, GaussNoise, Normalize
-
Resnet : 참가자 공유코드를 베이스라인으로 삼아 resnet18,resnet34,wide resnet 등 모델의 깊이를 변경하고 batch size,lr,epoch을 조정하며 train
-
Result
| parameters | model | learning rate | batch size | epoch |
|---|---|---|---|---|
| - | Wide ResNet50 | 0.01 | 128 | 50 |
| scores | train RMSE | dev RMSE | test RMSE |
|---|---|---|---|
| - | 30.5407 | 83.84183 | 78.944 |
- 다양한 시도
- densenet
- detectron2
- mask r-cnn
Click Here