# 模型文件说明
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”包,并且保证解压后直接得到所有相关文件(外层勿嵌套其他文件夹)。