[Python] YOLOV5 Window build with Anaconda

CUDA : 11.0

GPU : RTX 3060

 

1. Conda로 가상환경 설정 (YoloV5는 Python 3.7이상 지원)

conda create -n yolov5 python=3.8

2. Pytorch 설치 (YoloV5는 Pytorch 1.7.0이상 지원)

conda activate yolov5
conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=11.0 -c pytorch

3. Yolov5 repo 가져오기 및 필요 모듈 설치

git clone https://github.com/ultralytics/yolov5
cd yolov5
pip install -r requirements.txt

 

- Demo : 제대로 설치되었는지 확인해보자!

python detect.py --weights yolov5s.pt --img 640 --conf 0.25 --source data/images

yolov5/runs/detect/exp에 yolov5s의 weight로 얻은 detection 결과 이미지를 얻을 수 있음!