5인 이상 집합한 경우를 탐지하는 object detection 프로젝트
https://www.youtube.com/watch?v=Q5vHOuMOPrs
테이블과 사람을 탐지 후 테이블 갯수에 따라 그룹을 생성합니다.
탐지된 사람은 유클리드 거리 기준 가장 가까운 테이블 그룹에 속하게 됩니다.
그룹의 인원이 5명 이상이 되면 화면의 경계선에 빨간색과 노란색 테두리가 생겨 경고합니다.
구글 이미지 검색에서 식탁과 사람 각 600장의 사진을 모아 학습시켰습니다.
데이터의 양이 적어 탐지율이 낮지만, 이는 데이터 양을 늘려 충분히 학습시킨다면 더 좋은 결과를 보일 것입니다.
사전에 학습된 가중치를 사용하기 위해 > model < 다운로드
본 소스는 hunglc007의 tensorflow-yolov4-tflite에 일부 창작 코드를 추가한 코드입니다.
original source : https://github.com/hunglc007/tensorflow-yolov4-tflite (hunglc007)
threshold = 0.7
input_size = 416
left_top_x = 200
left_top_y = 200
detect_width = 1280
detect_height = 720
model_path = 'models/'
class_info = 'obj.names' model = 'yolov4-0613' # 사용할 모델 'yolov4-first', 'yolov4-0613'