搭建 YOLO v5 環境

lyp_發表於2020-11-21

官方指導:

https://github.com/ultralytics/yolov5/blob/master/tutorial.ipynb

環境搭建:

conda install git
conda create -n yolov5 python=3.8
conda activate yolov5 
git clone https://github.com/ultralytics/yolov5.git
如果下載比較慢:
git clone https://github.com.cnpmjs.org/ultralytics/yolov5 # clone repo
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple -U -r requirements.txt  使用映象
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.1 -c pytorch
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple ipython
cd yolov5
pip install ipykernel
python -m ipykernel install --user --name yolov5

測試:
使用 預訓練 模型進行測試:

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

在這裡插入圖片描述