我是19届完全模型的,我们小队的模型训练是我做的,今天希望三文教会你怎么训练模型!别的模型训练也可以参考吖,我感觉都很适用。(主要是自己总结一下) 

本文特别感谢学长学姐,老师们,完模车友们,鼠鼠,的细心指导,谢谢大家!

目录

一.数据集制作

二.AIstudio

三.本地编译

3.1 docker

3.1.1下载docker

3.1.2导入镜像

以下为注意事项


一.数据集制作

智能车之完全模型——模型训练(数据集篇)-CSDN博客文章浏览阅读391次,点赞11次,收藏12次。我是19届完全模型的,我们小队的模型训练是我做的,今天希望三文教会你怎么训练模型!别的模型训练也可以参考吖,我感觉都很适用。(主要是自己总结一下) https://blog.csdn.net/tuyaOjimihUa/article/details/145753221

二.AIstudio

智能车之完全模型——模型训练(炼丹篇)-CSDN博客文章浏览阅读659次,点赞21次,收藏19次。这个文件如下图所示,只需把数据标签改成自己需要的类就行了。 https://blog.csdn.net/tuyaOjimihUa/article/details/145735357

三.本地编译

3.1 docker

3.1.1下载docker

Windows用户安装第一步开启Hyper-V。

同时按下windows+r,打开运行对话框,输入appwiz.cpl,回车打开程序与功能的窗口,然后一下步骤开启。

如果遇到windows功能中没有Hyper-V选项(windows11家庭版就没有),需要自己下创建。可以参考:https://blog.csdn.net/youeyeb/article/details/144905924

开启后,去docker官网下载。我第一次下载docker就失败了,原因是版本太新||_||,所以大家不要下载最新版本的,会有bug。(本来想资源绑定的,可是博主还不会搞,显示已有该资源就失败了,想要资源的可以私我~)

安装完成打开终端,查询docker版本,如下图就说明安装好了。

然后按照提示或者查阅相关资料安装wsl2。

3.1.2导入镜像

在文件夹中创建名为work的文件夹并打开

打开终端后输入: docker pull pphub/ppnc2.0:v1.0(拉取的是外网的镜像,最好用一下梯子,如果不用下一天都下不完而且会经常断掉出错)

下载完成后输入以下指令:

docker images //查看镜像

docker run -it --cap-add=SYS_PTRACE --name ppnc2.0_docker 

-v D:\work:/home/edgeboard/workspace pphub/ppnc2.0:v1.0 /bin/bash & //将主机的work文件夹和eg板的workpace文件夹进行映射

这一步可能会遇到映射失败,不过没关系只要自己把文件上传到docker镜像的相关文件中就行,然后后面的一些步骤本来是在本地进行的,需要移到docker镜像的终端进行!!!

exit //退出容器

docker start ppnc2.0_docker //启动容器

docker exec -it ppnc2.0_docker /bin/bash  //进入容器

source /opt/compiler/scripts/activate_env.sh //导入环境变量

mv /home/compiler/ /home/edgeboard/workspace //把compiler文件夹映射到文件中(如果需要compiler文件也可以私我~)

cd /home/edgeboard/workspace/compiler //进入compiler

python3 compile.py ./config.json //开始编译

上面是必要性的指令,不过还要对镜像的文件进行处理:

①  在compiler文件中创建yolov3文件夹,再在yolov3文件夹中分别创建model和image文件夹。其中,model文件夹中放的是在ai四丢丢上下载下来的的模型,image文件夹中放的是本地编译需要训练的照片,50张!!!(这50张照片中要包含每个标签类型)

② 对compiler文件中的config.json文件进行配置:

照片尺寸大小可以用软件netron来看(也有网页版)

③  对compiler文件中的下 split.py文件的第 292 进行配置:

输出量也需要用netron查询。

④修改compile/res/文件夹下 io_paddle.json 文件中的 shape 参数

这个参数只需和上面的照片大小保持相同。

编译完成之后,如果映射成功会在本地电脑上出现新文件,若没有成功,就需要去docker的文件夹中找。

以下为注意事项

1.不同的照片大小做出来的模型大小不同,从而影响摄像头帧率

2.照片大小是可以在训练模型的时候直接修改,不需要手动

3.修改了照片大小后部署到eg板上后,需要对代码进行修改,不匹配就会报错

内容参考北京赛曙科技有限公司。感谢管看

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐