Kospo X Dacon Car Crash Analysis AI Contest(Recruiting)

Recruit | Algorithm | Video | Vision | Classification | Macro F1 Score

 

비디오 이미지 변환 코드

2023.02.21 04:02 2,192 Views language

대회 데이터를 라벨별로 폴더에 넣어주고 이미지로 바꿔주는 코드입니다.
코드를 실행할 때마다 dataloader을 만드는 과정에서 매번 컨버트 하는게 시간이 많이 소요되어 아예 따로 이미지로 저장해서 학습하려고 만들었습니다.
test셋에 적용할 때 라벨을 하드코딩 해놔서  폴더가 0부터 12까지 자동으로 생깁니다.
이미지로 바꾸는 과정에서도 비디오의 라벨을 찾으면서 dataframe 전체를 훑는데 비효율적입니다. df의 크기가 크지 않아서 마찬가지로 적당히 놔뒀습니다.
이외에도 더 부족한 점이 많을겁니다.
잘하시는 분이 보면 부끄러울 수준이라
불편하시면 적당히 수정하거나 필요한 부분만 뺴가시면 좋을 것 같습니다.
여담으로 py 파일은 업로드 지원 안하는게 불편하네요...
제 경우엔 전체 파일 변환하는데 30~40분 걸린 것 같습니다.

Code