Go to file
guyiyun 63d8466c43 📜 docs: add notes for a new version 2025-08-01 17:42:42 +08:00
LICENSE Initial commit 2025-02-19 15:48:23 +08:00
README.md 📜 docs: add notes for a new version 2025-08-01 17:42:42 +08:00

README.md

modelzoo

免责声明

  1. 本仓库仅作为研究、参考,若要实际使用,请务必遵守原算法工程的开源规则。
  2. 本仓库仅作为模型部署参考,大部分模型前后处理的迁移等运行时代码做了精度测试并与原算法做了对比,但未做全面的功能测试验证。请勿未经验证直接将本仓库代码直接用于实际项目,本仓库 不承担任何不良后果

请在上方选取您安装icraft版本所对应的模型库分支配套使用。每个分支都会新增更多模型以下为稳定版本对应分支推荐您使用最新的icraft版本和模型库分支

本仓库主要放一些已经适配、跑通的模型。

每个模型都有全流程使用说明的readme文档里面记录了模型适配的状态。由于模型众多其源码和原始权重需要您根据说明下载特定版本。

版本更新说明:

modelzoo_v3.31.0

  1. 2_compile/config目录下新增子文件夹, 仅支持 编译 BY 文件夹下toml文件toml文件中在[parse]阶段新增配置:

    target = "buyi"
    
  2. 3_deploy/modelzoo/MODEL/cfg目录下的yaml文件中更新以下配置

    imodel:
      mmuMode: true					# 是否使用mmu模式
      ocm_option: 4					# 对应编译toml文件中ocmopt配置
      run_backend: buyi				# 指定运行后端
      dump_output: false			# 是否dump网络的输出特征图
      dump_format: SFB				# 指定dump特征图使用的格式
      log_path: ../io/output/ftmp	# 指定dump特征图的文件保存路径
    
    • mmuMode:取代原配置名mmu,参数要求无变化
    • run_backend:取代原配置名sim,表示运行后端, 仅支持 配置为hostbuyi,前者表示仿真运行,后者表示上板运行
    • dump_output新增配置指定是否dump输出网络的输出特征图
    • dump_format新增配置指定dump特征图使用的格式支持SFB/SFT/SQB/SQT/HQB/HQT
    • log_path新增配置指定dump特征图的文件保存路径
    • ocm_option:新增配置,当前run_backend配置下不生效

modelzoo_3.6.2

在3_deploy/modelzoo/model/cfg/model.yaml中添加了四个模式工程中均默认为以下配置

  mmu: true						# 是否使用mmu模式
  speedmode: false				# 是否打开运行时的合并算子
  compressFtmp: false			# 是否打开运行时的内存压缩
  cudamode: false				# 仿真是否使用cuda模式
  • 在icraft3.6中编译toml生成网络的json和raw时默认已经打开了算子合并和内存压缩运行这样的网络需要使用mmu模式因此mmu默认打开

  • 而运行时的算子合并与内存压缩,与编译时的算子合并与内存压缩,本质上是相同的功能,不能重复打开,因此默认关闭

  • 添加cudamode配置打开cudamode运行仿真会使用显卡进行计算大幅提升推理速度节约仿真精度测试时间

modelzoo_3.1.0

对3.0.1版本模型库中的一处bug修复和一处精度优化

  • bug修复使用hardnms时硬件能够处理的box存在最大数量限制3.0.1中软件未对此进行限制。3.1.0中软件添加框数量限制,以防止板子卡死
  • 精度优化3.0.1模型库yolo系列模型保存box时由于没有物理意义对框的坐标保存取了整数实际场景使用时不产生影响。但是进行测试时精度数值会低于将框的坐标像素保存成小数时的精度3.1.0中改为保存小数框坐标