중소규모 건설 현장을 위한 AI 기반 안전 관리 서비스

1. AI 모델을 이용한 인지하지 못한 위험 요소 감지
2. 번거로운 문서 작업 보조
3. 관리자와 근로자 간 원활하게 소통할 수 있는 환경 제공

AI hub에서 제공되는 건설 현장 위험 상태 판단 데이터를 활용

YOLOv5 classification 모델을 활용한 분류 모델링 시행

1. [정상/비정상] 이진 분류

2. [추락/낙하/협착/화재/전도] 5대 사고 부류

정상/비정상 판별 후 비정상 판별 시 5대 사고 분류 예측

모델링 파일 onnx 로드 후 실제 데이터를 이용하여 예측 시 과적합 발생

YOLOv5 predict 이용 시에는 정확한 예측 결과, onnx 로드 시 성능 문제 발생

=> 문제 해결을 통한 보다 더 정확한 실제 데이터 예측 결과 도출

pretrained 모델 가중치를 이용한 학습 시 정상 데이터 성능 저하 발생

=> 데이터 추가 학습을 통한 클래스별 성능 개선

시간 관계 상 validation data 사용 (약20GB)

=> 제공 데이터(약200GB)를 모두 활용한 학습을 통한 성능 향상 기대