YOLO v5 installation 설치하기

YOLO v5 설치 매뉴얼

• Ubuntu
• 아나콘다

설치

#가상환경만들기
# 또는 파이참에서 새 프로젝트
conda create -n yolov5 python=3.8
conda activate yolov5


# yolo v5 가상환경안에 다운받기
# code clone
git clone https://github.com/ultralytics/yolov5.git

# if your yolov5 folder is lock (UBUNTU)
sudo chmod 777 -R ~/yolov5


### 아나콘다 네비게이터에서
# update
conda update -yn base -c defaults conda

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

# install pytorch
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
# 또는
pip3 install torch==1.8.1+cu102 torchvision==0.9.1+cu102 torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

# Extra
conda install -c conda-forge onnx

에러 모음집

이 외에도 여러 에러가 발생할 수 있지만,,,만나지 못했다.

ERROR : 추가로 opencv 설치 필요

pip install opencv-python

detect.py 가 있는 폴더에서 yolo가 잘 작동하는지 확인하기

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

train 에는 pycocotools 필요함

• 윈도우에는 지원 안함

pip3 install "git+https://github.com/philferriere/cocoapi.git#egg=pycocotools&subdirectory=PythonAPI"



pip install thop
pip install wandb