hailo-ai/hailo_model_zoo
⭐⭐⭐☆☆ (3/5)
🧩 软硬件结合
已发布
项目简介
The Hailo Model Zoo includes pre-trained models and a full building and evaluation environment
Hailo Model Zoo 是 Hailo 公司推出的一款面向高性能深度学习应用的开源模型库,旨在帮助开发者快速在 Hailo 系列 AI 加速芯片上部署和优化神经网络模型。该项目提供了大量预训练模型,涵盖图像分类、目标检测、语义分割、实例分割以及文字识别等主流计算机视觉任务,模型格式包括 ONNX、TensorFlow 以及可直接在 Hailo 设备上执行的 HEF 二进制文件。技术栈方面,项目基于 Python 3.10 至 3.12,依赖 TensorFlow 2.19.1 和 CUDA 12.5.1,并深度集成 Hailo Dataflow Compiler 5.3.0 与 HailoRT 5.3.0 运行时环境。核心功能上,用户不仅可以直接使用预训练模型进行精度评估,还能通过 Hailo 模拟器验证量化后的模型精度,最终生成高效的 HEF 文件,从而显著缩短从模型开发到硬件部署的周期。此外,项目还提供了详细的重新训练指南,支持用户在自定义数据集上对模型进行微调,满足特定业务场景的需求。Hailo Model Zoo 主要解决了边缘端 AI 推理中模型移植难、优化门槛高的问题,适用于智能安防、自动驾驶、工业质检、智慧零售等对实时性和能效比有严格要求的场景。无论是算法工程师还是嵌入式开发者,都能借助该工具链快速获得经过硬件适配的高质量模型,加速产品落地。