知识蒸馏3:YOLOV5项目准备

  • Post author:
  • Post category:其他




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 版权协议,转载请附上原文出处链接和本声明。