【实践经验】华为云AI应用在线服务部署custom_service写法

Kross Sun

已于 2023-01-10 11:17:15 修改

阅读量585

点赞数

分类专栏： AI算法常用技术文章标签：人工智能华为云 python

于 2023-01-05 23:44:15 首次发布

本文链接：https://blog.csdn.net/S_Kross/article/details/128572026

版权

前言

在参加华为云AI大赛的过程中，经常遇到的就是要编写custom_service.py, 今天来总结一下编写这个文件的一些注意事项。下面给出了一个例程进行分析。

例程

# !/usr/bin/python
# -*- coding: UTF-8 -*-

import json
try:
    from model_service.pytorch_model_service import PTServingBaseService
except:
    PTServingBaseService = object
    
import torch
import os
import numpy as np
from model import MyModel
import pathlib


class RadioMapService(PTServingBaseService):
    def __init__(self, model_name, model_path):
        print('--------------------init--------------------')
        self.model_name = model_name
        self.model_path = model_path
        print(f"model_name:{model_name}")
        print(f"model_pa