YOLOv5 개발환경 구축하기

어찌됬던 YOLOv8모델로 작업을 진행하려 했으나 TFLITE 변환에 실패하고 말았다.

YOLO 공식사이트에서도 TFLITE 변경을 지원하는 정식 버전을 YOLOv5로 안내하는 것으로 보아 v5가 모바일에 탑재하기 가장 적합하다는 판단을 했다.

완전 초보자도 함께 진행 할 수 있도록 아주 쉽게 설명해놓도록 하겠다.

 

 

먼저 https://github.com/ultralytics/yolov5 사이트에 방문하여 다운로드를 진행하자.

 

 

파이썬에서 가상환경을 만들고 난 뒤 git clone을 통해 소스코드를 받기로 하겠다.

 

아무 폴더나 만들고 여기서 커맨드창을 실행시킨다.

 

 

위에 cmd 만 입력하고 엔터를 치면

 

커맨드창이 생성된다.

 

 

python -m venv yolov5

 

라고 입력한다.

 

가상환경을 설정하는 이유는 오래된 프로젝트 일 수록 라이브러리 충돌이 발생할 확률이 높기 때문이다.

 

 

잘 생성되었다. 이제 VScode를 활용해서 환경을 열어보자.

 

vscode를 실행시키고.

 

잘열리게 되었다.

 

이제 터미널을 열고 가상환경을 활성화 해보자.

 

 

생성된 터미널에서 아래와 같이 넣고 실행하면,

Scripts\activate.bat

 

실행 터미널 앞쪽에 (yolov5)라는 우리가 venv를 설정했을때 지정한 이름이 나온다.

 

이상태에서 gitclone을 수행하보자.

git clone https://github.com/ultralytics/yolov5.git

 

주소는 아래에서 복사한다.

 

 

그러면 yolov5폴더가 생성되고 아래 파일이 다운로드 받아진것을 확인할 수 있다.

 

여기서 requirements.txt 를 열어보면 이 소스코드를 실행하기 위한 라이브러리가 정리되어있다. 

pip install 명령을 통해 요구되는 라이브러리를 설치해보자.

 

cd yolov5
pip install -r requirements.txt

 

 

잘 설치 하고 있는 중이다.

많은 라이브러리를 설치해야 하기 때문에 시간이 오래걸릴 수 있으니 천천히 기다려보자.

 

설치가 완료된 모습이다.

제대로 설치되었는지 확인해보자.

 

pip list

 

 

잘 설치된 것을 확인할 수 있다.

 

이것으로 YOLO5 설치를 완료했다.

 

다음 포스팅에서는 예제 파일을 돌려보고 이후에는 tflite로 변환하는것을 진행해보겠다.