[사물인터넷 캡스톤 디자인] 폭력감지 CCTV 프로젝트 일지 ⑤

프로젝트 개요

어린이집 폭력감지 CCTV로, yolov7을 사용하여, 교사가 아이를 학대할 경우 부모님 핸드폰으로 해당 폭력 영상을 전송하여, 어린이집 학대 사건이 일어났을 경우 증거인멸을 하지 못하게 하며, 빠른 시일 내에 부모님이 아이가 학대받는지 아닌지 알 수 있도록 하기 위해 이번 프로젝트를 진행하게 되었습니다. 거꾸로 어린이집 교사 입장에서는 아이가 작은 생채기가 생겨도 교사의 학대에 의한 것이 아닌, 아이의 실수에 의해 상처가 났다는 등 억울한 일이 생기지 않도록 도와줄 수 있습니다.

프로젝트 현위치

저는 Yolov7의 pre-trained된 pose-estimation을 활용하여 keypoint값을 추출한 후, 해당 값을 LSTM에 학습시켜 폭력 감지 하는 딥러닝 파트를 담당하여 진행하고 있습니다. 현재 keypoint값을 추출하였고, LSTM 초기 학습은 성공하여, 추가 데이터로 학습에 진행중입니다.

오늘의 이슈

내일 최종 발표의 Q&A를 준비하며 LSTM 학습에 쓰인 내부 function들에 대해 조금 구체적으로 알아보자.

Activation function

tanh 함수

tanh는 Hyperbolic Tangent Function의 줄인 버전으로 쌍곡선 함수이다. sigmoid 함수를 변형하여 얻었다.

함수의 중심점을 0으로 옮겨, sigmoid함수를 사용할 때 중심이 0이 아니기 때문에 학습이 느려질 수 있다는 단점을 보완하였다.

 

softmax 함수

softmax 함수는 입력받은 값을 0~1사이의 값으로 출력해며, 출력값들의 총합을 1이 되도록 하는 함수이다. 딥러닝 모델의 경우 결과가 확률로 출력되는 경우가 대다수이기 때문에 보통 출력단의 Dense layer에서 softmax 함수를 사용하여 최종 결과를 도출해낸다. 그렇기 때문에, 우리의 프로젝트에서도 softmax함수를 사용하여 출력값을 도출하였다.

Loss Function

categorical cross entropy

categorical cross entropy의 경우, softmax activation의 뒤에 cross entropy를 붙이는 형태로 많이 표현합니다. 시간이 부족한 관계로 cross entroy에 대한 자세한 공부는 방학으로 미뤄두도록 하겠습니다..

이번 방학에 진짜 한다