广告

后摩尔时代的创新:在米尔FPGA上实现Tiny YOLO V4,助力AIoT应用

2024-12-04 米尔电子 阅读:
学习如何在 MYIR 的 ZU3EG FPGA 开发板上部署 Tiny YOLO v4,对比 FPGA、GPU、CPU 的性能,助力 AIoT 边缘计算应用。(文末有彩蛋)

Tip:学习如何在 MYIR 的 ZU3EG FPGA 开发板上部署 Tiny YOLO v4,对比 FPGA、GPU、CPU 的性能,助力 AIoT 边缘计算应用。(文末有彩蛋)FlVednc

FlVednc

为什么选择 FPGA:应对 7nm 制程与 AI 限制FlVednc

在全球半导体制程限制和高端 GPU 受限的大环境下,FPGA 成为了中国企业发展的重要路径之一。它可支持灵活的 AIoT 应用,其灵活性与可编程性使其可以在国内成熟的 28nm 工艺甚至更低节点的制程下实现高效的硬件加速。FlVednc

米尔的 ZU3EG 开发板凭借其可重构架构为 AI 和计算密集型任务提供了支持,同时避免了 7nm 工艺对国产芯片设计的制约。通过在 ZU3EG 上部署 Tiny YOLO v4,我们可以为智能家居、智慧城市等 AIoT 应用提供高效的解决方案。FlVednc

FlVednc

FlVednc

CPU GPU FPGA 架构对比FlVednc

了解 Tiny YOLO 模型及其适用性FlVednc

YOLO(You Only Look Once)是一种实时物体检测模型,它通过一次性扫描整个图像,实现高效的对象识别。FlVednc

而其简化版 Tiny YOLO v4 更适合嵌入式设备,具有较少的层数和参数。其轻量化特性更适合在资源受限的设备上运行,尤其在低功耗、实时检测的边缘计算设备中表现出色。FlVednc

相比传统 GPU,FPGA 能在小面积和低功耗下实现类似的推理性能,非常契合 AIoT 应用。像米尔 ZU3EG 这样的 FPGA 开发板,通过底板和丰富接口的载板设计,非常适合高效的嵌入式低功耗数据处理。FlVednc

FlVednc

Yolo V4 网络结构图FlVednc

FlVednc

Tiny Yolo V4 网络结构图FlVednc

(通过优化网络结构和参数,保持较高检测精度的同时,降低模型的计算量和内存占用)FlVednc

获取数据集和模型FlVednc

可下载开源训练集或预训练模型。为了确保兼容性,建议将模型转换为 ONNX 格式,以便后续能在 FPGA 上完成优化。FlVednc

下载 Tiny YOLO v4 模型:从 Darknet 的 GitHub 仓库 获取 Tiny YOLO 的预训练权重,或者在 COCO 等数据集上自行训练模型。自定义的模型适用于特定应用场景(如车辆检测、人脸检测等)。FlVednc

数据准备:若要自定义模型,可使用 LabelImg 等工具对数据集进行标注,将数据转为 YOLO 格式。之后,可将 YOLO 格式转换为 ONNX 格式,以便兼容 FPGA 优化工具链。FlVednc

FlVednc

Tiny YOLO 在 Darknet 上训练的截图FlVednc

通过 Vivado HLS 为 FPGA 准备模型FlVednc

要将模型部署到 FPGA,需要将神经网络操作转换为硬件级描述。使用 Xilinx 的 Vitis HLS(高级综合)可以将 Tiny YOLO v4 的 C++ 模型代码的转化为 Verilog RTL(寄存器传输级)代码,从而将模型从软件世界带入硬件实现。FlVednc

详细步骤FlVednc

模型层映射和优化FlVednc

将 YOLO 的每一层(如卷积层、池化层)映射为硬件友好的 C/C++ 结构。例如,将卷积映射为乘累加(MAC)数组,通过流水线实现并行化。FlVednc

算子加速与指令优化FlVednc

流水线(Pipelining):利用流水线来处理多项操作并行,减少延迟。FlVednc

循环展开(Loop Unrolling):展开循环,以每周期处理更多数据,尤其在卷积操作中有效。FlVednc

设置 DATAFLOW 指令,使层间独立处理。FlVednc

量化与位宽调整FlVednc

将激活值和权重量化为定点精度(例如 INT8),而非浮点数。这在维持准确度的同时显著降低计算量,尤其适合 FPGA 的固定点运算支持。FlVednc

FlVednc

Tiny YOLO 模型在 Vivado HLS 中的层层转化流程图FlVednc

使用 Vivado 综合与部署 Verilog 到 米尔的ZU3EG FPGA开发板FlVednc

当 HLS 生成的 RTL 代码准备就绪后,可以使用 Vivado 将模型部署到 FPGA。FlVednc

Vivado 中的设置FlVednc

将 HLS 输出的 RTL 文件导入 Vivado。FlVednc

在 Vivado 中创建模块设计,包括连接AXI 接口与 ZU3EG 的 ARM 核连接。FlVednc

I/O 约束与时序FlVednc

定义 FPGA 的 I/O 引脚约束,以匹配 ZU3EG 板的特定管脚配置。配置时钟约束以满足合适的数据速率(如视频数据 100-200 MHz)。FlVednc

进行时序分析,确保延迟和响应速度达到实时要求。FlVednc

生成比特流并下载到 ZU3EGFlVednc

生成的比特流可以直接通过 JTAG 或以太网接口下载到 ZU3EG。FlVednc

FlVednc

将 Tiny YOLO 处理模块连接到 米尔ZU3EG开发板 的外设和接口FlVednc

在 FPGA 上测试并运行推理FlVednc

现在 Tiny YOLO 已部署,可以验证其实时对象检测性能。FlVednc

数据采集FlVednc

通过连接的相机模块捕捉图像或视频帧,或者使用存储的测试视频。FlVednc

使用 ZU3EG 的 ARM 核上的 OpenCV 对帧进行预处理,再将它们传入 FPGA 预处理后进行推理。FlVednc

后处理与显示FlVednc

模型检测对象后,输出边框和类别标签。使用 OpenCV 将边框映射回原始帧,并在每个检测到的对象周围显示类别和置信度。FlVednc

性能测试FlVednc

测量帧速率(FPS)和检测准确度。微调量化位宽或数据流参数,以优化实时需求。FlVednc

FlVednc

Tiny YOLO 模型在 ZU3EG 上显示检测结果的实时输出,视频帧中标注了检测到的对象FlVednc

性能优化与调试技巧FlVednc

为提高性能,可以进行以下调整:FlVednc

内存访问:设计数据存储方式,最大限度利用缓存并减少数据传输,降低内存瓶颈。FlVednc

降低延迟:重新评估关键路径延迟。若延迟过高,调整 Vitis HLS 中的流水线深度,并验证层间的数据依赖性。FlVednc

量化改进:尝试 INT8 量化。Xilinx 的 Vitis AI 可帮助微调量化参数,以平衡准确性与速度。FlVednc

FlVednc

不同优化配置对资源使用的影响FlVednc

FlVednc

图:米尔MYC-CZU3EG/4EV/5EV-V2核心板及开发板FlVednc

在MYIR 的 ZU3EG 开发平台上提供了一种高效的解决方案。利用 FPGA 独特的灵活性和低功耗优势,助力未来 AIoT 设备的普及和智能升级。FlVednc

关注米尔电子公众号,后台回复FPGA,获取完整Tiny Yolo V4教程。FlVednc

FlVednc

责编:Demi
文章来源及版权属于米尔电子,EDN电子技术设计仅作转载分享,对文中陈述、观点判断保持中立,不对所包含内容的准确性、可靠性或完整性提供任何明示或暗示的保证。如有疑问,请联系Demi.xia@aspencore.com
  • 微信扫一扫
    一键转发
  • 最前沿的电子设计资讯
    请关注“电子技术设计微信公众号”
广告
广告
热门推荐
广告
广告
EE直播间
在线研讨会
广告
面包芯语
广告
向右滑动:上一篇 向左滑动:下一篇 我知道了