YOLOX 기반 제조현장 위험 객체 인식 및 실시간 탐지 모델 개발
Development of a YOLOX-Based Hazardous Object Recognition and Real-Time Detection Model for Manufacturing Sites
1. 프로젝트 개요
프로젝트 요약
본 프로젝트는 제조현장의 작업자와 안전 관련 객체를 실시간으로 인식하기 위한 YOLOX 기반 객체 탐지 모델을 개발하는 것을 목표로 한다. 작업자(Person)를 비롯하여 안전모(Helmet), 안전조끼(Vest), 보호안경(Goggles), 장갑(Gloves), 안전화(Boots) 등의 개인보호구(PPE)를 학습하여 제조현장 영상 또는 이미지에서 각 객체의 위치와 종류를 실시간으로 탐지한다. 학습 데이터 전처리, 객체 탐지 데이터셋 구축, YOLOX 모델 학습 및 성능 개선, 정량적 성능평가를 순차적으로 수행하고, 최종적으로 Streamlit 기반 웹 데모를 구현한다. 사용자가 이미지 또는 영상을 입력하면 탐지된 객체의 Bounding Box, 클래스, Confidence Score와 분석 결과를 직관적으로 확인할 수 있도록 구성하여 제조현장 안전관리 AI의 실제 적용 가능성을 검증한다
핵심 키워드
2. 프로젝트 목표 및 내용
배경 및 필요성
제조현장에서는 작업자의 안전모, 안전조끼, 보호안경 등 개인보호구 착용 여부와 작업자 주변의 위험 요소를 지속적으로 확인해야 한다. 그러나 관리자가 CCTV 또는 현장을 직접 확인하는 기존 방식은 다수의 작업구역을 동시에 감시하기 어렵고, 반복적인 모니터링 과정에서 위험 상황을 즉시 발견하지 못할 가능성이 있다.
최근 딥러닝 기반 객체 탐지 기술이 발전하면서 영상 내 작업자와 안전 장비를 자동으로 인식하고 위험 상황을 실시간으로 분석할 수 있게 되었다. 특히 제조현장과 같이 객체의 크기, 작업자의 자세, 조명, 촬영 거리 및 배경이 지속적으로 변하는 환경에서는 정확성과 실시간성을 동시에 확보할 수 있는 객체 탐지 모델이 필요하다.
본 프로젝트에서는 실시간 객체 탐지에 적합한 YOLOX를 활용하여 제조현장의 작업자 및 PPE 객체 탐지 모델을 구축한다. 단순 모델 학습에 그치지 않고 데이터 전처리부터 모델 성능평가, 실제 영상 추론 및 Streamlit 기반 시각화까지 구현함으로써 향후 스마트팩토리 안전관리 시스템 및 산업현장 영상 분석 서비스에 활용할 수 있는 기반 기술을 확보하고자 한다.
프로젝트 목표
정량적 목표
- 제조현장 안전 관련 6종 이상의 객체 클래스 인식 모델 구축
- Person
- Helmet
- Vest
- Goggles
- Gloves
- Boots
- 객체 탐지 성능 mAP@0.5 90% 이상을 목표로 모델 학습 및 개선
- mAP@0.5:0.95, Precision, Recall 및 클래스별 AP/AR을 이용한 정량적 평가
- 실제 이미지 및 영상에서 Bounding Box와 객체 클래스를 동시에 탐지
- GPU 환경 기준 실시간 추론이 가능한 수준의 처리속도 확보
- 추론 지연시간 50 ms 이내 수준을 목표로 성능 검증
- 학습/검증/테스트 데이터셋을 분리하여 모델의 일반화 성능 검증
정성적 목표
- 제조현장 특성을 고려한 안전 객체 탐지 데이터 처리 파이프라인 구축
- YOLOX 구조 및 객체 탐지 모델의 학습·추론 과정 이해
- 데이터 품질 및 클래스별 탐지 난이도에 따른 성능 차이 분석
- 오탐(False Positive) 및 미탐(False Negative) 사례 분석을 통한 모델 개선
- 제조현장 영상에 적용 가능한 실시간 안전 객체 인식 기술 확보
- 사용자가 별도 AI 개발환경 없이 결과를 확인할 수 있는 Streamlit 기반 데모 시스템 구축
최종 결과물 형태
데모 시스템
3. 시연 계획
이 프로젝트는 최종 발표에서 실물 또는 조작 가능한 형태의 라이브 시연을 수행합니다. 화면 출력만으로 끝나는 결과물이 아니라, 실제로 만들어 동작시키는 것을 목표로 하는 팀을 모집합니다.
실물 시연 형태
최종 시연 시나리오
최종 시연에서는 Streamlit 기반 제조현장 안전 객체 탐지 웹 애플리케이션을 실행하여 모델의 실제 활용 과정을 보여준다. 먼저 사용자가 웹 화면에서 제조현장 이미지 또는 작업 영상 파일을 업로드한다. 업로드된 데이터는 학습된 YOLOX 모델에 입력되며 작업자와 Helmet, Vest, Goggles, Gloves, Boots 등의 안전 관련 객체를 자동 탐지한다. 탐지가 완료되면 원본 영상 위에 Bounding Box, 객체명, Confidence Score를 실시간으로 표시한다. 동시에 Streamlit 화면의 분석 영역에서는 탐지된 객체 수, 클래스별 탐지 결과 등을 표시하여 사용자가 제조현장의 안전 관련 객체 현황을 쉽게 확인할 수 있도록 한다. 시연은 다음 순서로 진행한다. 1. Streamlit 웹 애플리케이션 실행 - 프로젝트 및 모델 소개 - 탐지 가능한 객체 클래스 안내 2. 제조현장 이미지 입력 - 작업자가 포함된 테스트 이미지 업로드 - YOLOX 객체 탐지 수행 3. 객체 탐지 결과 시각화 - Person, Helmet, Vest, Goggles, Gloves, Boots 등의 Bounding Box 표시 - 객체별 클래스명 및 Confidence Score 출력 4. 동영상 기반 실시간 탐지 - 제조현장 작업 영상 업로드 - 프레임 단위 객체 탐지 수행 - 작업자의 이동 및 PPE 객체를 연속적으로 탐지 5. 탐지 결과 분석 - 전체 탐지 객체 수 - 클래스별 탐지 수 - Confidence 결과 - 모델 처리시간 및 추론속도 표시 6. 실제 성공/실패 사례 비교 이를 통해 단순히 학습 성능 수치를 제시하는 것을 넘어 실제 제조현장 영상이 입력되었을 때 AI 모델이 안전 관련 객체를 자동으로 인식하고 결과를 사용자에게 제공하는 전체 흐름을 시연한다.
시연 장소 / 필요 장비·재료
캡스톤디자인2 강의실
중간 점검 시연 시점
중간 점검에서는 데이터셋 구축 및 전처리 결과와 YOLOX 기반 1차 객체 탐지 모델을 시연한다. 테스트 이미지에 대해 작업자 및 PPE 객체의 Bounding Box, 클래스, Confidence Score가 정상적으로 출력되는 것을 확인하고, 초기 mAP 및 클래스별 탐지 성능을 제시한다.
4. 학부생 팀원 모집 공고
모집 분야/역할
모델 개발 2명, 데모 구현 1명
필요 역량/선수과목
우대 사항
머신러닝·딥러닝에 대한 기본적인 이해가 있는 학생
참여 시 혜택
실제 공개 산업안전 데이터셋을 활용한 객체 탐지 프로젝트 전 과정 경험, 프로젝트 결과물을 개인 포트폴리오 및 취업·진학 자료로 활용 가능, 논문 저자 등록, 특허 출원, SW 등록
지원 방법/문의
플랫폼 지원 및 개별 연락
5. 학기별 추진 계획
| 기간 | 단계 | 주요 활동 |
|---|---|---|
| 1~2주 | 팀 빌딩 & 주제 구체화 | 팀원별 역할을 분담하고 제조현장 안전관리 문제를 정의한다. 탐지 대상 객체와 YOLOX 기반 개발 범위를 확정하고 프로젝트의 정량적 목표와 최종 Streamlit 시연 형태를 구체화한다. |
| 3~5주 | 문헌조사 & 데이터 준비 | YOLO 계열 및 제조현장 PPE Detection 관련 연구를 조사한다. 공개 제조·산업안전 데이터셋을 확보하고 Annotation 구조를 분석한 후 Train/Validation/Test 데이터셋을 구축한다. |
| 6~9주 | 설계 및 1차 구현 (중간발표) | YOLOX를 기반으로 객체 탐지 모델을 구성하고 pretrained model을 활용하여 1차 학습을 수행한다. mAP, Precision, Recall 및 클래스별 AP를 분석하고 테스트 이미지 객체 탐지 결과를 구현하여 9주차 중간 발표에서 시연한다 |
| 10~13주 | 개선 및 실험/검증 | 데이터 증강, 학습 파라미터 조정, 입력 해상도 및 Confidence/NMS threshold 등을 실험한다. 오탐과 미탐 사례 및 클래스별 성능을 분석하여 모델을 개선하고 실제 제조현장 이미지·영상에서 일반화 성능을 검증한다. |
| 14~16주 | 결과 정리 & 발표 (최종발표) | 최종 모델의 mAP, Precision, Recall, FPS 및 추론시간을 정리한다. Streamlit 기반 웹 데모를 구축하고 이미지·동영상 객체 탐지 및 결과 시각화 기능을 통합한다. 최종 보고서, 발표자료 및 시연 영상을 제작하여 최종 발표를 진행한다. |
정기 미팅 계획
온/오프라인
역할 분담 방식
데이터 전처리, 모델 학습
6. 기대 효과 및 활용 계획
학습 효과
프로젝트 참여자는 실제 객체 탐지 프로젝트의 전 과정을 경험함으로써 컴퓨터 비전 및 딥러닝 실무 역량을 확보할 수 있다. 데이터셋 분석, Annotation 변환, 데이터 전처리, YOLOX 모델 학습, Hyperparameter 조정, 정량적 성능평가, 오류 분석 및 추론 최적화까지 수행하면서 객체 탐지 시스템 개발 과정을 체계적으로 이해할 수 있다. 또한 Streamlit을 이용해 학습된 AI 모델을 웹 서비스 형태로 구현함으로써 단순 모델 개발을 넘어 AI 모델과 응용 서비스 간 연동 경험을 확보할 수 있다. 팀 단위 프로젝트 수행을 통해 역할 분담, 실험 관리, 결과 분석 및 기술 발표 능력 또한 향상할 수 있다.
향후 활용 계획
논문/포트폴리오(특허 출원, SW 등록)