개발자 놀이터

1. 서론 YOLOv4는 Darknet 기반으로 구현했으나 필자는 코드를 이해하기 어려워 Pytorch로 커스터마이징 된 코드를 사용하였다. 2. 설치 1. git을 사용하여 코드를 다운로드한다. git clone https://github.com/WongKinYiu/PyTorch_YOLOv4.git 2. pytorch 설치, cuda, cuDNN 설치 windows10: https://jeo96.tistory.com/entry/Pytorch-CUDA-cuDNN-%EC%84%A4%EC%B9%98-windows-10 Pytorch, CUDA, cuDNN 설치 [windows 10] 1. 그래픽 카드 확인 1. 상태바에서 마우스 우클릭 2. 작업 관리자 클릭 3. 성능 클릭 4. GPU 클릭 GPU를 간단히..

1. 서론 이번 글에서는 커스텀 데이터를 사용하는 가이드를 작성해보도록 하겠다. https://jeo96.tistory.com/entry/YOLOv5-%EC%84%A4%EC%B9%98-Pycharm YOLOv5 설치 [Pycharm] 1. 서론 대학원에 입학하고 벌써 1년동안 YOLOv5로 연구하고 논문을 쓰니 약간 정이든 느낌이 든다. 처음에 설치환경 구성하기 쉬워서 사용하였지만 나름 다른 모델에 뒤쳐지지 않는 모델중 하나 jeo96.tistory.com 2. 본론 2.1 커스텀 데이터 구성 2.1.1 파일 이미지 1장과 txt 파일 1개가 쌍을 이루어야 하며 파일 이름은 이미지 이름과 레이블 이름이 같아야 하고 레이블은 txt 파일로 구성해야 한다. 2.1.2 label 포맷 yolov5의 학습 데이..

1. 서론 대학원에 입학하고 벌써 1년 동안 YOLOv5로 연구하고 논문을 쓰니 약간 정이든 느낌이 든다. 처음에 설치환경 구성하기 쉬워서 사용하였지만 나름 다른 모델에 뒤처지지 않는 모델 중 하나인 것 같다. 2. 본론 python은 www.python.org에서 python을 다운로드한 환경으로 가정하고 설명을 진행하겠다. 필자는 Anaconda사용에 익숙하지 않고 Pycharm에 익숙하다. 2.1 설치 구글에 yolov5 검색 후 깃허브 접속 git을 설치하여 cmd로 설치하는 방법과 zip 파일을 다운로드하여서 설치하는 방법이 있다. 초보자를 기준으로 zip 파일로 다운로드하는 경우로 설명하겠다. 파일을 다운로드하고 압축 해제한다. 폴더를 우클릭하여 PyCharm으로 연다. 우클릭으로 PyChar..