Yolov5安装教程

文章首发及后续更新:https://mwhls.top/2319.html
新的更新内容请到mwhls.top查看。
无图/无目录/格式错误/更多相关请到上方的文章首发页面查看。

基于NAIE平台的YOLOv5识别超超超可爱的逢坂大河

目录
1. 实现目标
2. 资源下载
3. 文件整理
4. 库安装
5. 运行时出错
6. 测试运行
7. 后记

实现目标

  • 出现如图1时,表示能正常运行,如果预测不出结果,说明是参数设置太低了。
    • 训练教程参考:https://blog.csdn.net/ai_faker/article/details/107099907
    • 最开始我用5次迭代花了半小时训练的,即图1,能出结果,但是不能预测任何东西。
    • 然后用20次迭代,花了一小时预测出来了,理论值见图2,预测值见图3。
图1:跑5次迭代的结果 图2:test_batch2_labels.jpg,20次迭代时的理论图 图3:test_batch2_pred.jpg,20次迭代后的预测图

资源下载

  • 我使用的yolov5:GitHub wudashuo/yolov5YOLOv5 汉化版,保持同步更新
    • yolov5官方github:https://github.com/ultralytics/yolov5
    • yolov5官方使用教程:https://github.com/ultralytics/yolov5/wiki/Train-Custom-Data
  • python版本:
    • Anaconda Windows 64位最新版(截止2021/5/6)。
      • 即python3.8.5,64位。
      • 最开始是用的最新的Python3.9.4,但很多库安装太麻烦了,所以改用了Anaconda。
  • coco128数据集:
    • 官方下载地址:https://www.kaggle.com/ultralytics/coco128
    • 百度网盘:https://blog.csdn.net/ai_faker/article/details/107099907
      • 这篇博客有提供网盘下载地址。

文件整理

  • 下载好yolov5之后解压在任一文件夹中。
  • 将coco128数据集放在yolov5数据集所在文件夹。
  • 文件树型结构如下:
    • 父文件夹
      • coco128
      • yolov5

库安装

  • 使用requirements.txt文件安装。
    • cmd工作目录切换至yolov5,输入:
    • pip install -r requirements.txt
    • 等待自动安装。
  • 出错:
    • 安装时大概率会出错,从出错信息往上找,找到正常输出信息中,最后一个安装的包
    • 开头为collecting xxx…
    • 百度python 安装 xxx 出错
  • scipy安装出错:
    • 需要先安装numpy与mkl
      • 下载地址:https://www.lfd.uci.edu/~gohlke/pythonlibs/#numpy
      • 我使用的是python3.8.5 x64,所以下载numpy‑1.20.2+mkl‑cp38‑cp38‑win_amd64.whl
        • 其中,cp38表示python3.8版本,win_amd64表示64位。
    • 然后在刚刚numpy的下载地址里面找到scipy,根据Python版本下载。
      • 或者直接使用requirements.txt安装,应该能正常安装,我的torch是这么安装上的。
    • 安装:
      • cmd工作目录切换至下载目录。
      • pip install 文件名
      • 例如,安装numpy:
        • pip install 'numpy‑1.20.2+mkl‑cp38‑cp38‑win_amd64.whl’
        • 不确定是不是长这样,因为我是按tab键自动补齐的。

运行时出错

  • 一般是已经安装好的库出了问题。
    • 需要卸载重装这个库。
      • pip uninstall xxx
      • pip install xxx
    • 把报错信息的最后几行复制下来百度即可。
    • 可能出错的库:Pillow, PyYAML
  • ‘xxx’ distribution错误:
    • pip install xxx
    • 或者卸载重装。
  • error: Microsoft Visual C++ 14.0 or greater is required. Get it with “Microsoft C++ Build Tools”: https://visualstudio.microsoft.com/visual-cpp-build-tools/:
    • 安装VC++环境。
    • 如果装上去还不行,试试装个VS2017,推荐用完整包装。

测试运行

  • 打开cmd,工作目录转至yolov5文件夹,输入:
    • python train.py
    • 如果库都安装成功,这段代码能正常执行,并开始输出,结果见文章开头。
  • 注:
    • 本文不提供参数修改的教程,参数修改可以参考这篇:https://blog.csdn.net/ai_faker/article/details/107099907

后记

  • 在将yolov5迁移至华为NAIE中时,不论如何修改,它始终提醒
    • AssertionError: Image Not Found coco128\images\train2017\000000000009.jpg
    • 即便将yaml文件的路径改了也这么提示。
    • 即便完全copy到NAIE中,也这么提示。
    • 注:解决了,删除coco128中的cache文件即可。

你可能感兴趣的:(python,python,YOLO,机器学习)