矩池云上安装yolov5并测试教程
官方仓库:https://github.com/ultralytics/yolov5
官方文档:https://docs.ultralytics.com/quick-start/
此案例我是租用了k80 镜像选择的是pytorch1.8.1
克隆仓库
git clone https://github.com/ultralytics/yolov5.git
这个地方我用了国内的镜像加速
https://github.com.cnpmjs.org/ultralytics/yolov5.git
安装需要的环境
cd yolov5
pip install -r requirements.txt
测试
python detect.py --source data/images/ --weights yolov5s.pt --conf 0.4
以我为例可以在/yolov5/runs/detect/exp5
目录下看到测试的图片
参考文章
YOLOv5—RTX 3080 conda环境配置cuda11.1、cudnn8.0.5、pytorch1.8.0、tensorrt7.2.2.3