YOLO v5 Pytorch 깃헙 링크 : https://github.com/ultralytics/yolov5
1. 가상환경 생성
conda create -n yolov5 python=3.8
conda activate yolov5
2. 깃헙 클론
#code clone
git clone https://github.com/ultralytics/yolov5.git
3. 필요한 라이브러리 설치
yolo v5 깃헙의 requirements.txt 참고해서 설치
# update
conda update -yn base -c defaults conda
# install Lib for YOLOv5
conda install -c anaconda cython numpy pillow scipy seaborn pandas requests
conda install -c conda-forge matplotlib pyyaml tensorboard tqdm opencv
#install pytorch
#자신 pc에 맞는 파이토치 깔기
conda install pytorch torchvision torchaudio -c pytorch
#conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
# Extra
conda install -c conda-forge onnx
자신의 pc에 맞는 torch install version 체크 가능
'DL' 카테고리의 다른 글
텐서플로우를 활용하여 신경망 구현하기 - 모델 구현 (0) | 2022.11.24 |
---|---|
텐서플로우를 활용하여 신경망 구현하기 - 데이터 전 처리 (0) | 2022.11.24 |
퍼셉트론의 알맞은 가중치 찾기 (0) | 2022.11.24 |
DIY 퍼셉트론 만들기 (0) | 2022.11.24 |
퍼셉트론 작동 예시 구현하기 (0) | 2022.11.24 |