1) 下载项目文件
2) 安装所需库
使用清华镜像源:
在yolov5-6.1-distillation路径下执行:
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
注意:simple 不能少, 是 https 而不是 http
3) 下载预训练权重文件
yolov5的github 链接:
https://github.com/ultralytics/yolov5
下载
YOLOv5 v6.1
版本的
yolov5s.pt,yolov5m.pt,yolov5l.pt,yolov5x.pt
权重文件,并放置在weights文件夹下
4) 安装测试
测试图片:
在yolov5-6.1-distillation路径下执行
python detect.py --source ./data/images/ --weights weights/yolov5s.pt
版权声明:本文为weixin_38346042原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。