전북대학교 교내에서 촬영한 데이터셋을 이용해 YOLOv4 model 학습(using google colab)
레포지토리 내 yolov4 폴더는 4개의 custom file이 들어있습니다. (i.e. yolov4-custom.cfg, obj.data, obj.names and process.py) except obj.zip(labeled images).
obj.zip 파일은 yolo format으로 레이블링된 텍스트 파일을 포함합니다.
NOTE yolov4-custom.cfg, obj.data, obj.names 3개의 파일을 본인이 찾고자 하는 class에 맞게 수정하여 사용하시면 됩니다. (제 경우 helmet, no_helmet 2가지 class 사용)
https://colab.research.google.com/drive/16biTYL3mgmbJghKDcjzKArI81vnGuYZr#scrollTo=Py0DQRy4HHDH