Yolov5安装教程

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

实现目标

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

资源下载

• 我使用的yolov5：GitHub wudashuo/yolov5: YOLOv5 汉化版,保持同步更新
  • 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文件即可。