모션 키포인트 검출 AI 경진대회

[kcseo25] | Private 19등 | 15.46656 | DarkPose

2021.04.08 23:19 7,911 Views language

아쉽게도 순위권에는 들지 못했지만 스스로에게 공부가 많이 되는 시간이었습니다.
K-Fashion AI 경진대회에서 데이콘 측에서 베이스라인으로 올려주신 mmdetection이라는 toolbox를 활용해 봤고,
mmdetection를 만든 open-lab의 mmpose를 사용해 Keypoint Detection을 마저 진행했습니다.
toolbox들을 활용한 관계로 코드 상 직관적이지 못한 점이 다수 존재하더라도 이해해 주셨으면 합니다.

저의 경우에는, Bounding Box를 얻기 위해 DetectoRS 모델의 Object Detection을 먼저 진행했습니다.
특정 사진 (철봉에 매달린  사진, 노 젓는 기구(?)를 사용하는 사진)에서 만족스럽지 못한 결과가 나와 추가적으로 데이터를 AI Hub에서 구해 학습을 진행했습니다. 그리고 Keypoint Detection에는 데이터에 맞게 DarkPose를 수정해 각각의 Keypoint들을 구했습니다.

대회를 마치며 한가지 아쉬운 점은 최정명님께서 찾아주신 오류 인덱스들을 수정해놓고 DataFrame 조작 과정에서 저장 실수로 인해 결과에 반영을 못한 점입니다. 다음번 대회에는 toolbox가 아닌  torch쪽을 공부해 스스로 코드를 짜봐야겠습니다.

아래는 mmdetection, mmpose를 사용하기 위한 configs 및 data setting을 위한 파일들과 가중치 파일, requirements,txt 그리고 추가데이터 파일들이 있는 드라이브입니다. 해당 코드들은 Colab Pro 환경에서 작성했습니다.
share folder : https://drive.google.com/drive/folders/1jMmk4Jf26SsRRYc3hsPFbE7EO-0Rzn3t?usp=sharing

Code
로그인이 필요합니다
0 / 1000
도비콘
2021.04.08 23:20

수고하셨어요