# 模型文件说明

Octopus模型管理模块,支持您上传自定义推理模型,并将其用于自动标注或模型评估任务。任务执行仅通过调用启动文件进行推理,并不感知和干预自定义模型的加载使用,因此启动文件尤为关键,您可以结合实际所需进行编写。若需要将模型用于自动标注或模型评估,则必须在自定义模型包根目录下包含命名如下的启动文件:

  • customer_auto_label.py

    自动标注启动文件。若需要将模型用于自动标注任务,则模型包根目录下必须包含该文件。具体要求请参见自动标注启动文件

  • customer_inference.py

    模型评估启动文件。若需要将模型用于模型评估任务,则模型包根目录下必须包含该文件。具体要求请参见模型评估启动文件

# 前提条件

自定义模型包中的自定义模型及其相关脚本文件,均需基于Tensorflow或PyTorch引擎框架编写。

# 模型文件基本要求

自定义模型包“xxx.zip”中必须包含启动文件以及自定义模型文件。除此之外还可以包含一些其他必要的自定义脚本、自定义库、自定义算子库等。模型包文件解压后目录结构可参考如下:

.
 ├─ xxx                          #模型包根目录(名称可自定义)
     ├─ customer_auto_label.py    #“自动标注”启动文件(如该模型用于自动标注,则必选,名称固定) 
     ├─ 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        #自定义模型文件(必选,名称可自定义)
     └─ ...                       #其他(可选)

注意: 您需将模型全部相关文件压缩为“.zip”包,并且保证解压后直接得到所有相关文件(外层勿嵌套其他文件夹)。

  • 启动文件

    主要包含用户使用Python3.6内置方法,及Tensorflow或PyTorch原生算法库编写的函数方法;

  • 自定义库

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

  • 自定义脚本

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

  • 自定义模型文件

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

  • 自动标注启动文件

  • 模型评估启动文件

上次更新: 2020/12/25 16:26:48