Hailo 模型库
.. |python| image:: https://img.shields.io/badge/python-3.8%20%7C%203.9%20%7C%203.10-blue.svg :target: https://www.python.org/downloads/release/python-380/ :alt: Python 3.8 :width: 150 :height: 20
.. |tensorflow| image:: https://img.shields.io/badge/Tensorflow-2.12.0-blue.svg :target: https://github.com/tensorflow/tensorflow/releases/tag/v2.12.0 :alt: Tensorflow :width: 110 :height: 20
.. |cuda| image:: https://img.shields.io/badge/CUDA-11.8-blue.svg :target: https://developer.nvidia.com/cuda-toolkit :alt: Cuda :width: 80 :height: 20
.. |compiler| image:: https://img.shields.io/badge/Hailo%20Dataflow%20Compiler-3.28.0-brightgreen.svg :target: https://hailo.ai/company-overview/contact-us/ :alt: Hailo Dataflow Compiler :width: 180 :height: 20
.. |runtime| image:: https://img.shields.io/badge/HailoRT%20(optional)-4.18.0-brightgreen.svg :target: https://hailo.ai/company-overview/contact-us/ :alt: HailoRT :width: 170 :height: 20
.. |license| image:: https://img.shields.io/badge/License-MIT-yellow.svg :target: https://github.com/hailo-ai/hailo_model_zoo/blob/master/LICENSE :alt: License: MIT :width: 80 :height: 20
.. image:: docs/images/logo.png
|python| |tensorflow| |cuda| |compiler| |runtime| |license|
Hailo 模型库提供预训练模型,用于高性能深度学习应用。使用 Hailo 模型库,您可以测量每个模型的全精度准确性、使用 Hailo 仿真器的量化准确性,以及在 Hailo-8 设备上的准确性。最后,您将能够生成 Hailo 可执行格式(HEF)二进制文件,以加速开发并生成使用 Hailo-8 加速的高质量应用程序。Hailo 模型库还提供重新训练说明,以便在自定义数据集上训练模型,以及针对内部数据集特定用例训练的模型。
模型 Hailo 提供不同的预训练模型,格式为 ONNX / TF,以及预编译的 HEF(Hailo 可执行格式)二进制文件,可在 Hailo 设备上执行。
这些模型分为:
-
公共模型 - 在公开可用的数据集上训练。
-
Hailo-8 -
分类 <docs/public_models/HAILO8/HAILO8_classification.rst>
、目标检测 <docs/public_models/HAILO8/HAILO8_object_detection.rst>
、分割 <docs/public_models/HAILO8/HAILO8_semantic_segmentation.rst>
、其他任务 <docs/PUBLIC_MODELS.rst>
-
Hailo-8L -
分类 <docs/public_models/HAILO8L/HAILO8L_classification.rst>
、目标检测 <docs/public_models/HAILO8L/HAILO8L_object_detection.rst>
、分割 <docs/public_models/HAILO8L/HAILO8L_semantic_segmentation.rst>
、其他任务 <docs/PUBLIC_MODELS.rst>
-
Hailo-15H -
分类 <docs/public_models/HAILO15H/HAILO15H_classification.rst>
、目标检测 <docs/public_models/HAILO15H/HAILO15H_object_detection.rst>
、分割 <docs/public_models/HAILO15H/HAILO15H_semantic_segmentation.rst>
、其他任务 <docs/PUBLIC_MODELS.rst>
-
-
对于Hailo-15M -
分类 <docs/public_models/HAILO15M/HAILO15M_classification.rst>
、目标检测 <docs/public_models/HAILO15M/HAILO15M_object_detection.rst>
、分割 <docs/public_models/HAILO15M/HAILO15M_semantic_segmentation.rst>
、其他任务 <docs/PUBLIC_MODELS.rst>
-
对于Hailo-10 -
分类 <docs/public_models/HAILO10/HAILO10_classification.rst>
、目标检测 <docs/public_models/HAILO10/HAILO10_object_detection.rst>
、分割 <docs/public_models/HAILO10/HAILO10_semantic_segmentation.rst>
、其他任务 <docs/PUBLIC_MODELS.rst>
-
|
HAILO模型 <docs/HAILO_MODELS.rst>
_ 是在内部数据集上针对特定用例进行培训的。 | 每个Hailo模型都附有再训练说明。
再训练
Hailo还提供了再训练说明 <docs/RETRAIN_ON_CUSTOM_DATASET.rst>
_,用于使用自定义数据集训练Hailo模型库中的网络。
基准测试
| Hailo的基准测试列表可在 hailo.ai <https://hailo.ai/developer-zone/benchmarks/>
_ 找到。
| 如需复现测量结果,请参考以下页面 <docs/BENCHMARKS.rst>
_。
快速入门指南
-
安装Hailo数据流编译器并进入虚拟环境。如果您不是Hailo客户,请联系
hailo.ai <https://hailo.ai/company-overview/contact-us/>
_ -
安装HailoRT(可选)。仅在需要在Hailo-8上运行时才需要。如果您不是Hailo客户,请联系
hailo.ai <https://hailo.ai/company-overview/contact-us/>
_ -
克隆Hailo模型库
.. code-block::
git clone https://github.com/hailo-ai/hailo_model_zoo.git
-
运行安装脚本
.. code-block::
cd hailo_model_zoo; pip install -e .
-
运行Hailo模型库。例如,打印MobileNet-v1模型的信息:
.. code-block::
hailomz info mobilenet_v1
入门指南 ^^^^^^^^
有关完整功能,请参阅 安装指南 <docs/GETTING_STARTED.rst>
_ 页面(完整安装说明和使用示例)。Hailo模型库使用Hailo数据流编译器来解析、优化模型、模拟和编译深度学习模型。完整功能包括:
- | 解析:将输入模型转换为Hailo的内部表示。
- | 分析器:生成模型的分析报告。该报告包含有关您的模型和在Hailo硬件上的预期性能信息。
- | 优化:优化深度学习模型以进行推理,并生成输入模型的数值转换为压缩的整数表示。
| 更多信息请参阅我们的
优化 <docs/OPTIMIZATION.rst>
_ 页面。 - | 编译:运行Hailo编译器以生成可在Hailo硬件上执行的Hailo可执行格式文件(HEF)。
- | 评估:使用Hailo模拟器或Hailo硬件推理模型,并生成模型准确性。
有关Hailo数据流编译器的更多信息,请联系 hailo.ai <https://hailo.ai/company-overview/contact-us/>
_。
.. figure:: docs/images/usage_flow.svg
许可证
Hailo模型库根据MIT许可证发布。更多信息请参阅 LICENSE <https://github.com/hailo-ai/hailo_model_zoo/blob/master/LICENSE>
_ 文件。
联系方式
如需支持/请求/问题,请访问 hailo.ai <https://hailo.ai/>
_。
更新日志
更多信息请参阅我们的 更新日志 <docs/CHANGELOG.rst>
_ 页面。