更新时间:2024-10-30 GMT+08:00
分享

模型文件基本要求

模型文件通常包括模型图和模型权重文件,具体内容视用户算法决定,无其他要求。如果需要使用内置指标计算,则需满足以下条件:

自定义模型包中必须包含推理启动文件以及自定义模型文件,其中推理启动文件需要根据Octopus数据格式对推理结果存储为json进行适配,除此之外还可以包含一些其他必要的自定义脚本、自定义库、自定义算子库等。模型包文件目录结构可参考如下:

.
├─ model                          #模型包根目录,上传模型时进入该目录下(名称可自定义)  
      ├─ customer_inference.py     #“模型推理”启动文件(如该模型用于内置评测模板评测,需要提供该文件,名称可自定义)  
      ├─ customer_package          #自定义库(可选,名称可自定义)  
          ├─ __init__.py               #自定义库 - >  python库  
          ├─ tensor_define.py          #自定义库 - >  python库文件  
          └─ ...       
      ├─ customer_utils.py         #自定义脚本1(可选,名称可自定义)  
      ├─ customer_script2.sh       #自定义脚本2(可选,名称可自定义)  
      ├─ sub_directory             #子文件夹  
          ├─ customer_operation.so     #自定义算子库(可选,名称可自定义)  
          └─ coustomer_model.pb        #自定义模型文件(必选,名称可自定义)  
      └─ ...                       #其他(可选)
  • 启动文件。

    模型推理脚本,用户接收数据集路径和推理结果存放路径,按照一定要求将每张图片的推理结果存入对应路径json文件中。

  • 自定义库。

    允许用户使用自定义库,但不推荐使用需要编译的库,以避免与内置库文件冲突。示例中使用Python语言中的package作为自定义库。

  • 自定义脚本。

    允许自定义除启动文件以外的自定义脚本文件,可根据实际所需编写。

  • 自定义模型文件。

    自定义模型文件“xxx.pb”,需要通过编写自定义脚本加载并使用。自定义模型文件的存放位置及名称可自定义,可以将相关模型文件保存至子文件夹中,也可以保存至根目录下。

相关文档