当前位置:   article > 正文

yolov5知识蒸馏3:YOLOV5项目准备_yolov5知识蒸馏 github

yolov5知识蒸馏 github

1) 下载项目文件

2) 安装所需库

使用清华镜像源
yolov5-6.1-distillation路径下执行:

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
  • 1

注意: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/ 
    声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家自动化/article/detail/481075
    推荐阅读
    相关标签
      

    闽ICP备14008679号