目录
目录README.md

modelzoo

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

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

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

版本更新说明:

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中改为保存小数框坐标
邀请码