Project Icon

hailo_model_zoo

面向多AI任务的深度学习预训练模型集合

Hailo模型库提供针对分类、检测和分割等AI任务的预训练深度学习模型。用户可测量模型的全精度和量化精度,并生成用于Hailo硬件加速的HEF文件。该库还包含自定义数据集重训练指南和特定用例模型。支持快速上手、性能评估及Hailo硬件部署,助力高效AI应用开发。

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>_ 页面。

项目侧边栏1项目侧边栏2
推荐项目
Project Cover

豆包MarsCode

豆包 MarsCode 是一款革命性的编程助手,通过AI技术提供代码补全、单测生成、代码解释和智能问答等功能,支持100+编程语言,与主流编辑器无缝集成,显著提升开发效率和代码质量。

Project Cover

AI写歌

Suno AI是一个革命性的AI音乐创作平台,能在短短30秒内帮助用户创作出一首完整的歌曲。无论是寻找创作灵感还是需要快速制作音乐,Suno AI都是音乐爱好者和专业人士的理想选择。

Project Cover

有言AI

有言平台提供一站式AIGC视频创作解决方案,通过智能技术简化视频制作流程。无论是企业宣传还是个人分享,有言都能帮助用户快速、轻松地制作出专业级别的视频内容。

Project Cover

Kimi

Kimi AI助手提供多语言对话支持,能够阅读和理解用户上传的文件内容,解析网页信息,并结合搜索结果为用户提供详尽的答案。无论是日常咨询还是专业问题,Kimi都能以友好、专业的方式提供帮助。

Project Cover

阿里绘蛙

绘蛙是阿里巴巴集团推出的革命性AI电商营销平台。利用尖端人工智能技术,为商家提供一键生成商品图和营销文案的服务,显著提升内容创作效率和营销效果。适用于淘宝、天猫等电商平台,让商品第一时间被种草。

Project Cover

吐司

探索Tensor.Art平台的独特AI模型,免费访问各种图像生成与AI训练工具,从Stable Diffusion等基础模型开始,轻松实现创新图像生成。体验前沿的AI技术,推动个人和企业的创新发展。

Project Cover

SubCat字幕猫

SubCat字幕猫APP是一款创新的视频播放器,它将改变您观看视频的方式!SubCat结合了先进的人工智能技术,为您提供即时视频字幕翻译,无论是本地视频还是网络流媒体,让您轻松享受各种语言的内容。

Project Cover

美间AI

美间AI创意设计平台,利用前沿AI技术,为设计师和营销人员提供一站式设计解决方案。从智能海报到3D效果图,再到文案生成,美间让创意设计更简单、更高效。

Project Cover

AIWritePaper论文写作

AIWritePaper论文写作是一站式AI论文写作辅助工具,简化了选题、文献检索至论文撰写的整个过程。通过简单设定,平台可快速生成高质量论文大纲和全文,配合图表、参考文献等一应俱全,同时提供开题报告和答辩PPT等增值服务,保障数据安全,有效提升写作效率和论文质量。

投诉举报邮箱: service@vectorlightyear.com
@2024 懂AI·鲁ICP备2024100362号-6·鲁公网安备37021002001498号