YOLOv5 설치

 

YOLOv3 설치 시도 -> 자꾸 오류 떠서 실패..

 

# 가상환경에 yolov5 다운
git clone https://github.com/ultralytics/yolov5.git

# update
conda update -yn base -c defaults conda

# install library
conda install -c anaconda cython numpy pillow scipy seaborn pandas
conda install -c conda-forge matplotlib pyyaml tensorboard tqdm opencv

 

PyTorch 설치는

https://pytorch.kr/get-started/locally/ 

내 환경에 맞는 명령어를 입력해주면 된다.

난 맥북이니까

# install pytorch
conda install pytorch torchvision torchaudio -c pytorch

 

ONNX란? ONNX(Open Neural Network Exchange)는 그 이름에서 살펴 볼 수 있듯이, Tensorflow, PyTorch와 같은 서로 다른 DNN 프레임워크 환경에서 만들어진 모델들을 서로 호환되게 사용할 수 있도록 만들어진 공유 플랫폼이다.

# install onnx
conda install -c conda-forge onnx

 

이제 실행시켜보자

python detect.py --source ./data/images/bus.jpg

runs/detect/exp2 폴더에 들어가면 분석 결과가 저장되어 있다!!

 

우리 사진도 해봤다

ㅋㅋㅋㅋㅋㅋㅋㅋㅋㅋㅋㅋㅋㅋㅋㅋㅋㅋㅋㅋㅋㅋㅋㅋ

잘 된당