공부하는 펭구

YOLOv3 custom dataset을 가지고 훈련을 진행 해보도록 하겠다. 환경은 우분투 20.04 LTS 에서 진행 하도록 하겠다. 사전에 설치해야할 항목들은 1. Nvidia-driver 2. CUDA, CUDNN 3. OpenCV 4. Darknet 이렇게 준비가 되어 있어야 한다. 준비가 완료 되었으면 custom dataset을 준비해 준다. 필자는 YOLO_mark를 이용해 준비를 해둔 상태이므로 이부분은 본 게시물에서는 설명하지 않도록 하겠다. 준비가 되었으면 다음 단계로 진행한다. Darknet 폴더로 이동해 custom_yolov3 폴더를 만들어준다. 아래의 사진처럼 5개의 파일이 있어야 한다. 준비해야할 부분을 하나씩 보도록 하겠다. 1. img 폴더 폴더 내에는 yolo_mark를..

YOLOv3, YOLOv4를 학습 시키기 위해서는 Darknet을 사용해야한다. 따라서 Darknet설치를 진행해보도록 하겠다. Darknet을 설치 하기에 앞서 아래의 요건을 충족해야한다. 1. Nvidia-driver 설치 2. CUDA 설치, Cudnn 설치 3. OpenCV 설치 설치가 완료되었으면 이제 Darknet 설치를 진행한다. Darknet에서는 두가지 종류가 있다. 현재 github에 alexab의 Darknet과 pjreddie의 Darknet이 올라와 있다. alexab의 Darknet이 업데이트도 되어있고 편리기능이 추가되어서 사용하기 훨씬 좋기 때문에 alexab의 Darknet 설치를 진행한다. https://github.com/AlexeyAB/darknet GitHub - A..