资讯 社区 文档
技术能力
语音技术
文字识别
人脸与人体
图像技术
语言与知识
视频技术

Codelab Notebook自定义环境部署最佳实践

由于 Codelab Notebook 具备很强的开发灵活性,用户可能在 Codelab Notebook 开发各种各样的模型,为了保障用户自定义的模型可以顺利部署, Codelab Notebook 提供了自定义环境部署的功能。

注意:自定义环境部署功能仅支持通用型 Notebook 开发的模型,并且该流程仅支持公有云服务部署的方式。

本文介绍了从创建 Notebook 任务到训练模型,再到保存模型、部署模型的全流程。

从【飞桨精选案例】创建项目

1、在 BML 左侧导航栏中点击『飞桨精选案例』,注意查看开发语言和AI框架版本,后面需要选择对应的训练环境。

image.png

2、在『飞桨精选案例』中选中某一个感兴趣的项目,点击复制到我的项目,示例如下:

image.png

3、完成填写后点击下一步即可配置项目资源,支持选择训练环境和资源规格:

image.png

4、完成配置后点击确认并启动即可。

训练模型

1、点击进入项目查看案例内容。

image.png

2、案例中提供了配置依赖和下载数据集的链接,并且提供了训练代码,逐个 cell 点击运行即可完成整个案例的运行,并且可以得到训练出来的模型。

image.png

保存模型

1、进入到保存模型的路径(在训练的输出信息里面会显示模型保存的路径),将模型移动到下图中的 pretrainedmodel 文件夹下。

image.png

image.png

2、点击生成模型版本。

image.png

3、填写相关信息并选中模型文件。

image.png

image.png

image.png

4、生成的模型版本将会在『模型列表』中出现。

image.png

制作自定义环境

由于在模型的训练过程中安装了一系列依赖,所以需要在平台环境的基础上安装这些依赖制作自定义环境才可进行后续的部署动作。

1、点击左侧导航栏的『环境管理』,并且进入『自定义环境』,点击新建环境。

image.png

2、填写环境名称和描述。

image.png

3、点击新建版本。

image.png

4、选择基础环境,这里注意选择对应框架版本的CPU环境。由于本最佳实践所使用的是paddle2.2.2环境,所以选择paddle2.2.2的CPU环境即可。

image.png

5、填写环境内容,这里主要是填写该模型运行所需要的依赖项,依赖项可以查看原项目中的内容。

image.png

如下图,该项目中主要依赖requirements中的内容和filelock。

image.png

找到requirements中的内容加以复制,添加进环境内容中。(注意,环境内容的原理是和requirements文件一样的,在自定义环境的过程中,平台会依据环境内容通过pip install安装在这里列出的所有组件。)

image.png

本项目的示例依赖:

paddlex==2.1.0
tqdm
scipy
colorama
cython
pycocotools
visualdl >= 2.1.1
paddleslim == 2.2.1
shapely
paddlepaddle-gpu >= 2.2.0
opencv-python
lap
scikit-learn==0.23.2
motmetrics
matplotlib
chardet
openpyxl
filelock

6、填写环境验证命令,此处的原理是根据用户需要,填入一些命令来验证一些关键的包是否安装成功,可以按照示例填写也可以按需填写。点击确认即可进入构建流程。

image.png

image.png

7、构建和验证的过程时间较长,验证成功后即可开始使用。

image.png

配置模型

1、在 BML 左侧导航栏中点击『Codelab Notebook』,进入对应项目的『模型列表』。

image.png

2、找到对应的模型进行配置。

image.png

image.png

3、选择『仅在线服务』,选择对应的模型文件以及自定义的环境。

image.png

4、编辑服务代码,此处主要是加载模型和构造一个请求体(此处的请求体会影响后续部署为公有云服务后的调用代码,需要根据自身情况撰写),可以参考项目中的predict代码。

image.png

本项目的示例代码:

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# *******************************************************************************
#
# Copyright (c) 2022 Baidu.com, Inc. All Rights Reserved
#
# *******************************************************************************
""" custom predict code """
import base64
import numpy as np
import cv2
class PyModel(object):
    """ 模型服务预测封装类,支持用户实现自定义的服务请求数据处理和模型推理逻辑。"""
    def __init__(self, model_path, *args, **kwargs):
        """
        根据`model_path`初始化`PyModel`类,如加载标签文件、预处理参数、加载模型 等等
        :param model_path: 该目录下存放了在页面上选择的模型版本中包含的所有文件,并保留原有的目录结构;
        :param *args/**kwargs: 其他信息(预留字段,不可以删除),系统可能会在未来传入一些其他参数
        """
        # 加载模型
        import paddlex as pdx
        self._model = pdx.load_model(model_path + "/epoch_1/")
    def predict(self, request_body):
        """
        自定义对请求体的处理,包括请求处理、模型预测、预测结果封装等
        :param request_body: 调用接口的请求体json字典
        :return:
            response_body: 请求的处理结果,需要能够被json序列化
        """
        # 请求解析、模型推理、推理结果封装
        # 请求入参: {"image": image base64字符串}
        image = request_body["image"]
        image_bin = base64.b64decode(image)
        # 图像解码
        image = np.frombuffer(image_bin, dtype='uint8')
        image = cv2.imdecode(image, 1)
        # 模型推理计算,参考 https://github.com/PaddlePaddle/PaddleX/blob/develop/docs/apis/models/detection.md#predict
        result = self._model.predict(image)
        format_result = []
        for target in result:
            format_result.append({
                "label_index": target["category_id"],
                "name": target["category"],
                "score": target["score"],
                "location": {
                    "left": target["bbox"][0],
                    "top": target["bbox"][1],
                    "width": target["bbox"][2],
                    "height": target["bbox"][3]
                }
            })
        return {"results": format_result}
        
class CustomException(RuntimeError):
    """
    进行模型验证和部署服务必需的异常类,缺少该类在代码验证时将会失败
    在处理异常数据或者请求时,推荐在`PyModel`中的自定义推理predict函数中抛出`CustomException`异常,
    并为`message`指定准确可读的错误信息,以便在服务响应包中的`error_msg`参数中返回。
    """
    def __init__(self, error_code, message, orig_error=None):
        """ init with error_code, message and origin exception """
        super(CustomException, self).__init__(message)
        self.error_code = error_code
        self.orig_error = orig_error

5、点击保存和提交。

模型校验

1、配置完成的模型处于待校验状态,点击进行模型校验。

image.png

image.png

2、模型校验支持传入测试数据的base64格式的json(由于用户可能传入多种格式的数据,所以采用json格式传入更具有适配性)。

image.png

本项目中将对应的测试图片通过代码转换为了base64的json格式。

import numpy as np
import cv2
image_bin = open("/home/work/dataset/images/fire_000001.jpg", "rb").read()
import base64
print(base64.b64encode(image_bin))

示例:

数据类型 json格式
图片 {"image": "xx(此处粘贴转写为base64后的内容)"}

3、点击发起校验,即可查看预测结果,如果与预期一致,点击发布即可。

image.png

image.png

4、发布完成将会在模型管理中新增一行记录。

image.png

模型部署

1、点击即可进入部署阶段。

image.png

2、填写相关信息即可完成部署。

image.png

上一篇
基于 Notebook 的通用模板使用指南
下一篇
Notebook简介