MX-Yolov3本地训练模型的安装和使用

  • Maix的本地训练:不知道是不是我的问题,我来来回回重新装了好几遍,但是最后都会显示“model numpy”版本过低,但是升级“numpy”后,又与其他model不兼容,于是放弃。
  • MX-Yolov3:模型测试和模型转化[k210的模型是(.kmodel),但yolov3训练结果的模型是(.tflite),所以需要转化模型]
  • Maix的本地训练和MX-Yolov3并不兼容,前者需要 Python3.8 ,后者需要 Python3.7.4 ,而且MX-Yolov3安装时不能有其他Python版本。
  • MX-Yolov3训练结束后模型默认保存路径的错误,这个后面回写解决方法

首先感谢CSDN的大佬的帮助:

vott使用和数据集的制作请看这位大佬

Mx-yolov3环境配置+本地模型训练+K210

MaixHub在线训练的使用看这

K210 Mx-yolov3模型训练和物体识别

脚本运行和离线运行见这里

Mx-yolov3+Maixpy+ K210进行本地模型训练和目标检测

其实大佬解释的很详细。我写这篇文章主要是为了写下我遇到的问题和解决方法。

MX-Yolov3环境配置

自负武术

MX-Yolov3需要的Python环境是Python3.7.4版本,如果你的电脑里有其他Python版本,要先卸载干净

Mx-yolov3软件下载

链接:https://pan.baidu.com/s/1wWJT27TolPt8lh88Q1KDBQ 提取码:zfz6
安装程序末尾的目标文件夹包含环境配置所需的所有安装包

环境配置

到MX-yolov3的解压目录下,根据下图的路径进入
MX-Yolov3本地训练模型的安装和使用点击(使用前配置.exe),进行环境配置
MX-Yolov3本地训练模型的安装和使用之后就是环境配置。第一次,从左到右单击。

Python3.7.4安装

  1. 点击安装Python3.7.4会弹出python的安装界面,然后跟随视频
    【K210】识别神器Mx-yolov3安装教程。
    这里再次提醒各位,一定要把其他Python版本删除干净,不然下一步会出问题。
  2. 打开cmd,输入 Python 即可查看Python安装版本
    不出意外你会看到Python3.7.4
    MX-Yolov3本地训练模型的安装和使用

安装Python依赖库与预训练权重

重点!重点!重点! ! !
点击中间的(安装)按钮后,最好不要在你的电脑上进行其他操作,最好不要移动鼠标,否则你会发现你的配置软件会卡死。至于为什么,不知道哈哈哈哈,玄学

点击上面的(使用前配置.exe)图片的中间的按钮,会自动安装Python依赖库与预训练权重,运气好的话可以看到下面的图片(确实可以是说运气好,哈哈哈哈,我承认装这个非常艹)
MX-Yolov3本地训练模型的安装和使用

安装CUDA和cudnn

这一步其实成功与否对Mx-yolov3是否配置成功关系不大,这个是用来配置GPU,让GPU训练模型,安装失败顶多你的CPU会哭泣,问题不大,就训练慢一些。

前面说个这个Mx-yolov3文件夹内已有安装包,所以无需下载
根据这个一步步安装和添加环境变量,但是我们可以跳过下载那一部分Cuda和Cudnn 安装教程
打开cmd,测试CUDA是否安装成功

nvcc --version
nvcc -V

正确结果:
MX-Yolov3本地训练模型的安装和使用

环境配置失败解决方案

在下面的模型训练部分,您将知道您的环境配置是否成功。如果训练失败,很可能是你的环境配置有问题。
解决方案是:

  • 检查Python的版本,是否安装了其他Python版本
  • 重复点击(安装Python依赖库与预训练权重),说不定有奇效(玄学)
  • 删除已安装的文件夹并将其整体删除。如果不能删除,重启电脑,可以成功删除,然后重现安装配置环境。

环境培训

测试

准备

不要急于创建模型训练的数据集,可以使用文件夹中的现成数据集进行测试:

  • 将训练次数改为1,因为只是测试,所以不需要训练那么多次
  • 训练图片地址:Mx-yolov3_EN_1.2\datasets\yolo\masks\images
  • 训练标签地址:Mx-yolov3_EN_1.2\datasets\yolo\masks\xml
  • 点击自动提取
  • 点击计算Anchor
    如下所示
    MX-Yolov3本地训练模型的安装和使用

开始训练

如果前面都没问题的话,正常来说,训练速度会非常快的,而且会看到End of Training
MX-Yolov3本地训练模型的安装和使用训练结束后,自动弹出训练结果文件夹
MX-Yolov3本地训练模型的安装和使用

如果你看到上面的界面,就证明你已经起飞了,一起加油吧! ! ! ! !哦耶! ! !

模型测试

这里有一点!重点!重点! ! !

模型训练结束后不是弹出一个文件夹吗?它在 Mx-yolov3_EN_1.2\Model_file下,找到你刚刚训练结束的文件夹
将里面的文件全部复制粘贴到Mx-yolov3_EN_1.2\datasets\yolo\masks\Train_data这里
MX-Yolov3本地训练模型的安装和使用
如图所示
MX-Yolov3本地训练模型的安装和使用
点击模型测试,打开Mx-yolov3_EN_1.2\datasets\yolo\masks\Train_data,选择.h文件
过一会就会看到这个界面

MX-Yolov3本地训练模型的安装和使用

模型转换

K210用的是模型是.kmodel,MX-yolov3生成的是.tflite,所以需要转换模型

选择文件

MX-Yolov3本地训练模型的安装和使用

  • 模型输入路径:刚刚复制粘贴后的路径里的.tflite文件
  • 模型输出路径:选择刚才复制粘贴的路径
  • 量化图片路径:Mx-yolov3_EN_1.2\datasets\yolo\masks\Train_data\Test_img(就是模型里的测试集)

单击(转换)

如果不出意外,您将看到:
MX-Yolov3本地训练模型的安装和使用

成功就会在文件夹里看到.kmodel文件
MX-Yolov3本地训练模型的安装和使用成功! ! !

运行脚本

附带测试代码,在文件夹(程序)中
MX-Yolov3本地训练模型的安装和使用点击打开boot.py文件,查看要求的文件名
我记得需要把label.txt改名为classes.txt
将boot里用的文件放到SD卡里就完成了,over!!!

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
心中带点小风骚的头像心中带点小风骚普通用户
上一篇 2022年4月3日 下午7:31
下一篇 2022年4月3日 下午7:48

相关推荐