$ hub install yolov3_darknet53_venus==1.0.0

API

def context(trainable=True,
            pretrained=True,
            get_prediction=False)

提取特征，用于迁移学习。

参数

• trainable(bool): 参数是否可训练；
• pretrained (bool): 是否加载预训练模型；
• get_prediction (bool): 是否执行预测。

返回

• inputs (dict): 模型的输入，keys 包括 'image', 'im_size'，相应的取值为：
  • image (Variable): 图像变量
  • im_size (Variable): 图片的尺寸
• outputs (dict): 模型的输出。如果 get_prediction 为 False，输出 'head_features'、'body_features'，否则输出 'bbox_out'。
• context_prog (Program): 用于迁移学习的 Program.

def save_inference_model(dirname,
                         model_filename=None,
                         params_filename=None,
                         combined=True)

将模型保存到指定路径。

参数

• dirname: 存在模型的目录名称
• model_filename: 模型文件名称，默认为__model__
• params_filename: 参数文件名称，默认为__params__(仅当combined为True时生效)
• combined: 是否将参数保存到统一的一个文件中

依赖

paddlepaddle >= 1.6.2

paddlehub >= 1.6.0