/Darknet_YOLOv4-Sanhak_Project_2

2021 가을학기 산학프로젝트2 - 킥보드 사용자 헬멧 미착용 감지

Primary LanguageC++

Darknet YOLOv4 - Sanhak Project 2

멘토기업: 테크노니아

filename.py

  • 이미지파일 이름 변경시 사용

txtmake.py

  • 빈 txt파일 생성. labelimg 사용시 미리 빈 txt파일 생성하는게 편함

set-divide.py

  • 이미지 파일을 training set, valid set 으로 random 하게 나눠줌

imgaug.py

  • 이미지 augmemtation 에 사용

image_opencv.cpp

  • line 1000 ~ 1013
  • detection 발생 시 캡쳐 후 이미지 저장 부분 : 기존 코드를 현재 opencv 버전에 맞게끔 수정.

pyolo.py

  • opencv의 dnn모듈을 사용해 yolo로 single image detection

pycamyolo.py

  • pyolo에서 jetson nano의 cameara를 사용해 영상 detection
  • Jetson Nano 이미지에 기본적으로 설치되어 있는 opencv로 실행할 경우 오류가 나므로 높은 버전의 opencv를 설치해야함

trt_yolo_cam.py