树莓派 5 AI 套件(Hailo)示例

kuan_li_lyg 2024-07-01 08:31:02 阅读 71

系列文章目录

 


前言

76ce83830a0144499ecd88fb497a07f9.png

        欢迎访问 Hailo Raspberry Pi 5 示例库。本项目展示了在 Raspberry Pi 5 上演示 Hailo AI 处理器功能的各种示例。这些示例将帮助您开始在嵌入式设备上使用人工智能。更多信息,请访问 Hailo 官方网站和 Hailo 社区论坛。 

8a40bdee678248839c817dd282ff4b7d.png


 

一、如何设置 Raspberry Pi 5 和 Hailo-8L

        在本指南中,您将学习如何将 Raspberry Pi 5 与 Hailo-8L 人工智能加速器连接起来。

1.1 所需设备

树莓派 5树莓派 M.2 M-Key HATHailo8L M.2 模块(也支持 Hailo-8)散热垫(随套件提供)可选:散热片可选项 Raspberry Pi 官方摄像头(如摄像头模块 3 或高质量摄像头)可选:USB 摄像头 USB 摄像头

1.2 硬件

        在本指南中,我们使用了 Raspberry Pi 5 型号、官方 Active Cooler 和 27W USB-C 电源。我们建议使用官方 USB-C 电源,以确保电路板能够为 M.2 HAT 提供足够的电力。

36bf8e3562d8439d84cfb1831486e383.png

1.2.1 树莓派 M.2 密钥帽

        Raspberry Pi M.2 M 键 HAT 可与 Hailo-8L M.2 键 M 或 B+M 配合使用。(也支持 Hailo-8)。安装 M.2 模块时,请务必使用散热垫,以确保 M.2 模块和 HAT 之间的正常散热。如果您的项目封装在机箱中,请确保适当的通风以避免过热。如果需要,可为 Hailo-8L 模块添加散热片。有关如何安装 M.2 模块的详细说明,请参阅 Raspberry Pi 的官方指南。

abf91cfecbc8432fbb4e1decdf5316c7.png

1.2.2 Raspberry Pi 摄像头

        有关如何安装摄像头的说明,请参阅《Raspberry Pi 摄像头指南》。

1.3 软件

1.3.1 安装 Raspberry Pi 操作系统

        从此处为您的操作系统(Windows、macOS 或 Ubuntu)下载并安装最新的 Raspberry Pi Imager。

        选择 Raspberry Pi 5。

 

5dcd18f8df9b4c10b801ef55c51fe4c6.png

        选择 Raspberry Pi OS(64 位)。

 

a230c8065b354de9bda8da7c8fe84edd.png

1.3.2 更新系统

        启动 Raspberry Pi 5 进入图形环境,更新基础软件。为此,打开终端窗口并运行 

sudo apt update

sudo apt full-upgrade

         这将把系统更新到最新的 Raspberry Pi 内核,其中包括 Hailo 驱动程序支持。

1.3.3 将 PCIe 设置为 Gen3

        要使 Hailo 设备达到最佳性能,必须将 PCIe 设置为 Gen3。虽然可以选择使用 Gen2,但会降低性能。

        打开 Raspberry Pi 配置工具:

sudo raspi-config

        选择 "6 高级选项 "选项,然后选择 "A8 PCIe 速度 "选项。选择 "是 "启用 PCIe Gen 3 模式。点击 "完成 "退出。

        重新启动 Raspberry Pi。

sudo reboot

1.3.4 安装 Hailo 软件

        安装 Raspberry Pi AI 套件运行所需的所有软件。为此,请在终端窗口运行以下命令:

sudo apt install hailo-all

        这将安装以下软件组件:

Hailo 固件HailoRT 运行时软件。更多信息,请参阅 HailoRT GitHub Repo。Hailo TAPPAS 核心软件包: 作为 TAPPAS 软件库的衍生软件,该核心软件包包含 GStreamer 元素、后处理功能以及该软件库使用的其他工具。它对于在 Pi 上开发应用程序至关重要,但不包括独立应用程序。更多信息和文档请参见 Hailo TAPPAS GitHub。rpicam-apps Hailo 后期处理软件演示阶段。更多信息,请参阅 Raspberry Pi 官方示例。

        重新启动 Raspberry Pi。

sudo reboot

1.3.5 验证安装

        现在可以检查系统是否识别 Hailo 芯片:

hailortcli fw-control identify

        如果一切正常,输出应该是这样的:

Executing on device: 0000:01:00.0

Identifying board

Control Protocol Version: 2

Firmware Version: 4.17.0 (release,app,extended context switch buffer)

Logger Version: 0

Board Name: Hailo-8

Device Architecture: HAILO8L

Serial Number: HLDDLBB234500128

Part Number: HM21LB1C2LAE

Product Name: HAILO-8L AI ACC M.2 B+M KEY MODULE EXT TMP

        如果看不到此输出,请检查 PCIe 故障排除部分。

        运行以下命令测试 TAPPAS Core 的安装情况:

        Hailotools:(TAPPAS Gstreamer 元素)

gst-inspect-1.0 hailotools

        预期结果:

Plugin Details:

Name hailotools

Description hailo tools plugin

Filename /lib/aarch64-linux-gnu/gstreamer-1.0/libgsthailotools.so

Version 3.28.2

License unknown

Source module gst-hailo-tools

Binary package gst-hailo-tools

Origin URL https://hailo.ai/

hailoaggregator: hailoaggregator - Cascading

hailocounter: hailocounter - postprocessing element

hailocropper: hailocropper

hailoexportfile: hailoexportfile - export element

hailoexportzmq: hailoexportzmq - export element

hailofilter: hailofilter - postprocessing element

hailogallery: Hailo gallery element

hailograytonv12: hailograytonv12 - postprocessing element

hailoimportzmq: hailoimportzmq - import element

hailomuxer: Muxer pipeline merging

hailonv12togray: hailonv12togray - postprocessing element

hailonvalve: HailoNValve element

        Hailonet:(HailoRT 推理 Gstreamer 元素)

gst-inspect-1.0 hailo

        预期结果:

Plugin Details:

Name hailo

Description hailo gstreamer plugin

Filename /lib/aarch64-linux-gnu/gstreamer-1.0/libgsthailo.so

Version 1.0

License unknown

Source module hailo

Binary package GStreamer

Origin URL http://gstreamer.net/

hailodevicestats: hailodevicestats element

hailonet: hailonet element

synchailonet: sync hailonet element

3 features:

+-- 3 elements

        如果找不到 hailo 或 hailotools,请尝试删除 GStreamer 注册表:

rm ~/.cache/gstreamer-1.0/registry.aarch64.bin

        如果一切正常,就可以继续安装 hailo_rpi5_examples。

1.4 故障排除

        有问题?加入 Hailo 社区论坛的讨论。

1.4.1 PCIe 故障排除

        确保 PCIe 板和 M.2 模块连接正确。要测试系统是否识别 PCIe 板,请运行以下命令:

lspci | grep Hailo

        如果输出结果如下

0000:01:00.0 Co-processor: Hailo Technologies Ltd. Hailo-8 AI Processor (rev 01)

        然后系统会识别 PCIe 板。如果没有,请检查连接和电源,并确保 PCIe 已启用(请参阅 Raspberry Pi 文档)。如果电路板是新的,可能需要更新 Raspberry Pi 5 的固件。

1.4.2 驱动程序问题

        如果出现 "未安装 Hailo 驱动程序 "的错误提示,请确保内核版本大于 6.6.31。你可以运行以下命令获取内核版本:

uname -a

        如果内核版本低于 6.6.31,可能需要运行 apt update 和 apt full-upgrade 来更新内核。如果内核版本没有问题,请重启系统并重试。

1.4.3 已知问题

        以下问题应由 TAPPAS Core 安装 deb 处理,但如果遇到这些问题,可以手动修复。

1.4.3.1 PCIe 页面大小问题

        某些主机不支持某些 PCIe 描述符页面大小。如果出现如下错误

[HailoRT] [error] CHECK_AS_EXPECTED failed - max_desc_page_size given 16384 is bigger than hw max desc page size 4096

        确保/etc/modprobe.d/hailo_pci.conf 存在,并包含以下一行:

options hailo_pci force_desc_page_size=4096

        检查配置:

cat /etc/modprobe.d/hailo_pci.conf

# expected result:

options hailo_pci force_desc_page_size=4096

1.4.3.2 无法在静态 TLS 块中分配内存

        在某些情况下(尤其是 aarch64),你可能会遇到以下错误,导致某些 GStreamer 插件无法正确加载。错误信息如下

bash

(gst-plugin-scanner:67): GStreamer-WARNING **: 12:20:39.178: Failed to load plugin '/usr/lib/aarch64-linux-gnu/gstreamer-1.0/libgstlibav.so': /lib/aarch64-linux-gnu/libgomp.so.1: cannot allocate memory in static TLS block

        将此添加到 .bashrc 文件中即可解决此问题:

echo 'export LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libgomp.so.1' >> ~/.bashrc

        如果您已经遇到了这个错误,可以通过运行以下命令来修复:

export LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libgomp.so.1

rm ~/.cache/gstreamer-1.0/registry.aarch64.bin

二、Hailo RPi5 基本管道

        该资源库包含 Hailo RPi5 平台的基本管道示例。这些示例演示了物体检测、人体姿态估计和实例分割。您可以将这些应用作为自己项目的基础。

2.1 安装

        克隆资源库

git clone https://github.com/hailo-ai/hailo-rpi5-examples.git

        输入版本库目录:

cd hailo-rpi5-examples

2.1.1 环境配置

        要运行示例,应确保环境设置正确。我们使用 hailo-tappas-core pkgconfig 文件获取 Hailo 依赖项。

        你可以通过以下脚本进行设置。该脚本将设置所需的环境变量,并激活 Hailo 虚拟环境(如果不存在,则会创建)。

source setup_env.sh

2.1.2 安装要求

        确保已进入虚拟环境并运行以下命令:

pip install -r requirements.txt

2.1.3 资源下载 

./download_resources.sh

2.2 应用程序结构

2.2.1 用户自定义数据类(user_app_callback_class)

        该用户自定义类作为输入传递给回调函数,该函数在管道输出上运行。它用于主程序与回调函数之间的通信。它对 hailo_rpi_common::app_callback_class 进行了扩展,并可根据应用程序的具体情况定制附加变量和方法。

2.2.2 应用程序回调函数(app_callback)

        您应在此处添加代码。这是一个用户自定义函数,用于处理管道中的每个帧。它由管道中的 "identity_callback "元素调用,位于网络推理和后处理之后。作为该函数输入的 GStreamer 缓冲区包括作为 Hailo 元数据的网络输出和帧本身。每个示例都演示了如何为其任务解析特定的元数据。有关 Hailo 元数据对象的更多信息,请参阅 TAPPAS 文档中的 Hailo Objects API。

2.2.3 GStreamer 应用程序类(GStreamerApp)

        使用基本管道时无需更改。该类通过扩展 hailo_rpi_common::GStreamerApp 来设置 GStreamer 管道并处理事件和回调。应用程序可以通过重载 get_pipeline_string 函数来修改网络参数和管道。有关 TAPPAS 管道和元素的更多详情,请参阅 TAPPAS 文档。

2.3 目标检测示例

e55b44633ce84d87b39eb9f83ab70724.png

        本例演示对象检测。它默认使用 YOLOv6n 模型。它还支持 yolov8s 和 yolox_s_leaky 模型。它使用 Hailo 的 NMS(非最大抑制)层作为 HEF 文件的一部分,因此所有使用 NMS 编译的检测网络都可以使用相同的代码。

        要运行示例,请使用

python basic_pipelines/detection.py --input resources/detection0.mp4

        要关闭程序,请按 "Ctrl+C"。

        如需其他选项,请运行

python basic_pipelines/detection.py --help

        另请参阅使用不同输入源运行。

2.3.1 本例中的内容:

2.3.1.1 实例分割回调类:

        回调函数展示了如何从网络输出中获取实例分割元数据。每个实例都表示为带有掩码(HAILO_CONF_CLASS_MASK 对象)的 HAILO_DETECTION。如果设置了--use-frame(使用帧)标志,代码将解析掩码,并根据帧协坐标调整它们的大小和形状。代码会将它们的形状打印到终端。在用户缓冲区上绘制掩码是可能的,但由于性能原因,本示例中并未实现。

2.3.2 其他功能

        使用 --help 标志运行任何示例,即可查看所有可用选项。例如

python basic_pipelines/pose_estimation.py --help

# Example output:

usage: pose_estimation.py [-h] [--input INPUT] [--use-frame] [--show-fps] [--disable-sync] [--dump-dot]

Hailo App Help

options:

-h, --help show this help message and exit

--input INPUT, -i INPUT

Input source. Can be a file, USB or RPi camera (CSI camera module). For RPi camera use '-i rpi' (Still in Beta). Defaults to

/dev/video0

--use-frame, -u Use frame from the callback function

--show-fps, -f Print FPS on sink

--disable-sync Disables display sink sync, will run as fast possible. Relevant when using file source.

--dump-dot Dump the pipeline graph to a dot file pipeline.dot

        有关如何使用这些选项的更多信息,请参阅下文。

2.3.2.1 使用不同输入源运行

        这些示例默认使用 USB 摄像头(/dev/video0)运行。您可以使用 --input 标志更改输入源。要使用 Raspberry Pi 摄像头运行,请使用 --input rpi。(仍处于测试阶段)下面是几个示例:

python basic_pipelines/detection.py --input /dev/video2

python basic_pipelines/detection.py --input rpi

python basic_pipelines/detection.py --input resources/detection0.mp4

        注意:USB 摄像头不一定是 /dev/video0。您可以运行以下命令检查哪些视频设备可用:

ls /dev/video*

        您可以运行以下程序来测试摄像头是否正常工作:

ffplay -f v4l2 /dev/video0

        如果出现错误,请尝试其他设备,如 /dev/video2(可能是偶数)。

2.3.2.2 使用帧缓冲区

        有关使用帧缓冲器的示例,请添加 --use-frame 标志。请注意,提取视频帧并将其显示出来会降低应用程序的运行速度。这种实现方式未经优化,仅作为一个简单示例。也有可能在缓冲区中写入内容并替换管道中的旧缓冲区,但这样做效率不高。

2.3.2.3 打印帧频 

        要打印帧频,请添加 --print-fps 标志。这将把帧频打印到终端和视频输出窗口。

2.3.2.4 转储管道图:

        这对调试和理解管道非常有用。要将流水线图形转存到 dot 文件中,请添加 --dump-dot 标志。这将在 basic_pipelines 目录下创建一个名为 pipeline.dot 的文件。要安装该工具,请运行

sudo apt install graphviz

        下面是使用 --dump-dot 标记运行检测示例的完整示例:

python basic_pipelines/detection.py --dump-dot

# To visulaize the pipeline run:

dot -Tx11 basic_pipelines/pipeline.dot &

# To save the pipeline as a png run:

dot -Tpng basic_pipelines/pipeline.dot -o pipeline.png

        下面是检测管道图的输出示例:

2f5628f539784940b879e2976ae8fa35.png

        小贴士 右键单击图像,选择 "在新标签页中打开图像",查看完整图像。

2.3.3 姿势估计示例

a535b603044243ab82c7fd0ec579b44a.png

2.3.4 实例分割示例

ea999ffceaf84283891bcbe21d4c6eb1.png

 

2.3.4 CLIP 应用程序 - 即将推出

        CLIP(对比语言-图像预训练),使用 Hailo-8L 人工智能处理器预测实时视频帧上最相关的文本提示。

2.3.6 树莓派官方示例

2.3.6.1 rpicam-apps

        树莓派 rpicam-apps Hailo 后处理示例。这是 Raspberry Pi 使用集成到其 CPP 摄像头框架中的 Hailo AI 处理器进行 AI 后期处理的官方示例。有关如何使用 rpicam-apps 的文档,请点击此处。要运行 rpicam-apps 中的示例,请按照以下说明操作: 克隆 rpicam-apps 软件源,获取 JSON 配置文件

git clone --depth 1 https://github.com/raspberrypi/rpicam-apps.git

        将目录更改为 assets 文件夹并运行示例

cd rpicam-apps/assets/

# Run the example

rpicam-hello -t 0 --post-process-file hailo_yolov6_inference.json --lores-width 640 --lores-height 640

2.3.6.2 picamera2 - 即将推出

        Raspberry Pi picamera2 是基于 libcamera 的 Picamera 替代品,Picamera 是 Raspberry Pi 传统相机栈的 Python 接口。Picamera2 还提供了一个易于使用的 Python API。

2.4 故障排除和已知问题

        如果遇到任何问题,请在 Hailo 社区论坛上开票。论坛上有很多有用的信息,可能已经包含了问题的解决方案。

RPi 摄像头输入仍处于测试阶段。它可能不稳定,可能导致应用程序崩溃。帧缓冲区未优化,可能会降低应用程序的运行速度。以下是一个简单的示例。DEVICE_IN_USE() 错误。DEVICE_IN_USE() 错误通常表示 Hailo 设备(通常为 /dev/hailo0)当前正在被其他进程访问或锁定。这种情况可能发生在并发访问尝试期间,或者前一个进程没有干净利落地终止,导致设备处于锁定状态。有关详细信息,请参阅社区论坛主题。

        解决步骤:

识别设备: 通常情况下,Hailo 设备位于 /dev/hailo0。确保这是您设置的正确设备文件。查找使用该设备的进程: 运行以下命令,列出当前使用 Hailo 设备的进程:

sudo lsof /dev/hailo0

终止进程: 使用上一条命令输出中的 PID(进程 ID)来终止进程。用实际 PID 替换 <PID>。

sudo kill -9 <PID>

 



声明

本文内容仅代表作者观点,或转载于其他网站,本站不以此文作为商业用途
如有涉及侵权,请联系本站进行删除
转载本站原创文章,请注明来源及作者。