sol’s blog

@hemashreeraja23/ driver-distraction-detection-via-hybrid-CNN-LSTM

☑️ SFDDD 데이터셋 사용

 

📂 데이터 로드 및 전처리

📌 운전자 매핑 정보 csv 파일 로드

 

📌 이미지 데이터 로드 및 전처리

 

📌 데이터 무작위 셔플링

 

📌 훈련 및 테스트 데이터 분리

  • 이미지 데이터를 NumPy 배열로 반환 후 훈련/테스트 데이터로 분할
  • to_categorical()을 사용해 라벨을 one-hot encoding 변환
 

📂 모델 구축(Xception 기반 전이학습)

 

📌 Xception 모델 사용

 

추천 글

BlogPro logo
Made with BlogPro

태그