코스포 x 데이콘 자동차 충돌 분석 AI경진대회 채용

채용 | 알고리즘 | 동영상 | 비전 | 분류 | Macro F1 Score

상금 : 채용
964명 마감

연습

대회안내 데이터 코드 공유 토크 리더보드

제출

[Private 10위] Resnet3D 코드 공유

안녕해요

2023.03.16 13:41 1,408 조회 language

제출 시 모델을 저장 과정과 Random seed 고정 부분에서 실수가 있어 완벽히 복원 가능하지 않은 점 참고 바랍니다.
코드에서 특별한 점은 ASL을 loss function으로 사용했고, 동영상을 확인하며 labeling을 직접 다시 했다는 점입니다.

OS 및 GPU 정보는 다음과 같고,
OS: Ubuntu 18.04.4
GPU: NVIDIA GeForce RTX 3090

주요 library 버전은 다음과 같습니다.
torch == 1.11.0+cu113
torchvision == 0.12.0+cu113
pandas == 1.3.5
numpy == 1.21.6
scikit-learn == 1.0.2

모델 weight와 새로 labeling한 csv 파일 및 코드 첨부합니다.
https://drive.google.com/file/d/14gmrFY5p6nqtzqeC84-8OArfmPD53Ken/view?usp=sharing

PDF

코드