본문 바로가기
AI/AI 부트캠프

[AI 부트캠프] DAY 104 - CV 프로젝트 5

by HOHHOH 2023. 12. 16.

[오늘의 일지]

프로젝트 - 학습을 위한 충돌 영상 분류

[상세 내용]

학습을 위한 충돌 영상 분류

작업의 목표

- 어제 1차적으로 전체 데이터에서 사고 데이터만 분류를 한 뒤에 나머지 사고 데이터만으로 다시 나머지 3개의 기준으로 멀티라벨링을 할 수 있는 모델을 구축하기로 계획이 되어있었습니다. 그렇기 때문에 첫 번째 모델이 사고와 비사고를 잘 분류시키기 위해서 따로 확실한 사고 장면만을 가지고 있는 이미지가 필요한 상황이 된 것입니다.

 

다시 사고 이미지 추출

- 제가 데이터의 전처리와 분석 담당이기 때문에 사고 장면만을 따로 추출해야 했습니다. 처음에는 그냥 영상의 마지막 5 프레임정도만 추출하면 대부분 뽑을 수 있겠다고 생각했지만 생각보다 기준이 없어서 다시 사고영상만을 하나씩 확인하는 과정을 하게 되었습니다.

 

추출 방법

- 저는 영상의 마지막 20 프레임만을 우선적으로 추출한 뒤 명확한 사고 장면이 포함되어 있는 부분만 3 프레임씩 정제했습니다. 작업은 모두 수작업으로 진행이 되었기 때문에 시간이 생각보다 오래 걸렸습니다.  작업은 아래와 같이 수행했습니다.

추출 과정

 

[마무리]

 오늘은 1차로 사고와 비사고 분류 모델의 학습 데이터를 구축하는 과정을 정리해 보았습니다. 영상 데이터를 다뤄보기 전에는 잘 몰랐지만 학습을 위해서는 이미지를 추출한 뒤 모든 작업이 이루어져야 했기 때문에 생각보다 전처리나 분석 과정에서 수작업으로 해야 하는 부분이 많았던 거 같습니다. 그래서 아직 모델을 돌리는 과정은 못했지만 다음 주에는 어느 정도 모델을 돌릴 수 있으면 좋을 거 같습니다.

반응형

댓글