분석시각화 대회 코드 공유 게시물은
내용 확인 후
좋아요(투표) 가능합니다.
Private 4위, LightGBM + CLIP
여러 keypoint와 이미지로부터 여러 Feature를 extraction하고 최종적으로 LightGBM을 활용하여 모델을 만들었습니다.
이미지 feature를 임베딩하기 위해 CLIP을 활용하였습니다.
gesture_type, hand_type, pose_name을 순서대로 학습하여 예측한 type을 다음 모델의 input으로 사용하였습니다.
Ryzen 5 3600 + 2060 super 기준, 전처리 및 학습부터 예측까지 약 40분 정도 소요되며,
제 PC 환경 상 py 파일이 아닌 Jupyter notebook에서 실행 시 PyTorch DataLoader의 num_workers를 늘리면 문제가 발생하여 0으로 시행하였습니다. (참고: https://github.com/pytorch/pytorch/issues/51344)
이를 늘린다면 시간을 단축할 수 있습니다.
감사합니다.
DACON Co.,Ltd | CEO Kookjin Kim | 699-81-01021
Mail-order-sales Registration Number: 2021-서울영등포-1704
Business Providing Employment Information Number: J1204020250004
#901, Eunhaeng-ro 3, Yeongdeungpo-gu, Seoul 07237
E-mail dacon@dacon.io |
Tel. 070-4102-0545
Copyright ⓒ DACON Inc. All rights reserved