1. 环境准备:树莓派5B与AI_KIT的初次见面
拿到树莓派5B和那块小小的AI_KIT加速卡时,我的第一感觉是,这玩意儿真能跑YOLOv8?毕竟YOLOv8在PC上都得靠显卡撑着。但实测下来,这套组合拳的威力远超预期,平均134帧的流畅度,让很多嵌入式AI项目从“能跑”变成了“跑得飞起”。在开始折腾模型训练和转换之前,我们得先把硬件和基础环境给搭稳了。这个过程有点像给新电脑装系统,虽然步骤固定,但细节决定成败,一步踩坑可能就得折腾半天。
首先,你得确保手头的树莓派5B已经装好了官方的Raspberry Pi OS(64位版本)。我强烈推荐使用最新的Bookworm版本,它对AI_KIT的支持最好。然后就是主角AI_KIT了,它本质上是一块搭载了Hailo-8L AI加速芯片的M.2板卡,需要插入树莓派5B背面的那个M.2 Key E接口。安装的时候注意对齐金手指,轻轻推入,最后用附带的螺丝固定好就行,整个过程和装个固态硬盘差不多简单。
硬件装好,通电开机。接下来就是软件环境的搭建。这里有个非常关键的步骤,也是官方推荐的最省心方法:一键安装Hailo全家桶。打开树莓派的终端,输入下面这条命令,然后喝杯咖啡等着就行。
sudo apt update && sudo apt install hailo-all
这条命令会帮你安装好所有必需的组件,包括Hailo的内核驱动、固件、运行时库,以及非常重要的Tappas后处理库和rpicam-apps的Hailo增强版。后处理库是干嘛的?简单说,Hailo芯片吐出来的是原始的、编码过的检测框数据,需要Tappas这个“翻译官”把它转换成我们能看懂的边界框坐标和类别。而rpicam-apps的增强版,则让你能直接用树莓派的自带摄像头或者USB摄像头作为AI模型的输入源,非常方便。
安装完成后,我建议做个简单的健康检查。运行 hailortcli fw-control identify 这个命令,如果能看到你的Hailo设备信息(比如型号是HAILO8L),那就说明硬件识别和驱动加载都没问题了。有时候可能会遇到权限问题,记得把当前用户加入到 hailort 组里,命令是 sudo usermod -aG hailort $USER,然后注销重新登录一下才能生效。环境搭好了,我们才算拿到了进入嵌入式AI世界的门票。
2. 训练你自己的YOLOv8模型
环境就绪,接下来就是模型的“灵魂”——训练一个能识别你特定目标的YOLOv8模型。很多人觉得训练模型是算法工程师的活儿,离嵌入式开发很远。其实不然,现在借助Ultralytics框架,训练一个可用的模型门槛已经低了很多。我的经验是,数据质量比模型结构更重要。哪怕你用最小的YOLOv8n模型,只要数据够好、标注够准,在树莓派上也能有出色的表现。
第一步,准备数据集。假设你想做一个“安全帽检测”项目。你需要收集大量包含“戴安全帽”和“未戴安全帽”工人的图片。数量上,每个类别能有几百张高质量的图片就足以启动训练了。图片的拍摄角度、光照条件、背景尽可能多样化。标注工具我推荐用 Roboflow 或者 LabelImg。标注时,边界框要紧紧包裹住目标物体,类别标签要准确。标注完成后,通常我们会把数据集整理成YOLO格式,也就是每张图片对应一个 .txt 文件,里面记录着物体中心点坐标、宽高(都是归一化后的值)和类别索引。
第二步,搭建训练环境。如果你有带NVIDIA显卡的电脑,本地训练是最快的。如果没有,完全可以利用云端GPU,比如Google Colab的免费T4 GPU,或者Kaggle的P100,训练YOLOv8这种模型完全够用。在Colab上,安装Ultralytics包只需要一行命令:pip install ultralytics。然后把你的数据集上传到Google Drive或者直接上传到Colab的运行时环境中。
第三步


388

被折叠的 条评论
为什么被折叠?



