人工智能yolo使用keras搭建模型_使用tensorflow serving部署keras模型（tensorflow 2.0.0）...

最新推荐文章于 2021-09-12 15:32:57 发布

沙金锐

最新推荐文章于 2021-09-12 15:32:57 发布

阅读量365

点赞数

文章标签：人工智能yolo使用keras搭建模型

本文链接：https://blog.csdn.net/weixin_42316286/article/details/113628496

版权

本文介绍了如何使用Tensorflow 2.0.0将基于Keras的YOLO模型导出并部署到Tensorflow Serving中。在新版本中，Keras模型可以直接使用model.save()方法导出，简化了之前复杂的导出流程。接着，文章讲解了如何通过Docker部署模型，并提供了请求客户端的代码示例，以便测试模型的正确运行。最后，作者提到已将内容收录在其个人网站和B站主页上。

摘要由CSDN通过智能技术生成

Tensorflow 2.0.0出来后，1.x版本的API有些已经改变，19年年初写的这一篇《TensorFlow Serving + Docker + Tornado机器学习模型生产级快速部署》文章，在tf 2.0.0版本里面有较大的变动，另外Tensorflow官方也推荐大家使用tf.keras，因此本文将会教大家如何使用tensorflow serving部署keras模型，适用tensorflow 2.0.0以后的版本。注：下面“tensorflow serving”将会简写为“tfs”。

导出Keras模型

keras模型训练完毕后，一般我们都会使用model.save(filepath)储存为h5文件，包含模型的结构和参数，而我们需要把这个h5文件导出为tensorflow serving所需要的模型格式：

from keras import backend as K
from keras.models import load_model
import tensorflow as tf

# 首先使用tf.keras的load_model来导入模型h5文件
model_path = 'v7_resnet50_19-0.9068-0.8000.h5'
model = tf.keras.models.load_model(model_path, custom_objects=dependencies)
model.save('models/resnet/', save_format='tf')  # 导出tf格式的模型文件

注意，这里要使用tf.keras.models.load_model来导入模型，不能使用keras.models.load_model，只有tf.keras.models.load_model能导出成tfs所需的模型文件。导出的文件路径结构如下：

.
└── 0
    ├── assets
    ├── saved_model.pb
    └── variables
        ├── variables.data-00000-of-00002
        ├── variables.data-00001-of-00002
        └── variables.index

最大的改变是，以往导出keras模型需要写一大段定义builder的代码，如这篇文章《keras、tensorflow serving踩坑记》的那样，现在只需使用简单的model.save就可以导出了。当然以前这种写法应该还能继续使用，能自定义signature、input、output的名称，但为了简单起见（用keras不就是为了简单嘛），直接一键导出更舒服。

导出以后，我们在终端执行以下命令，查看模型的signature、input、output的名称，后面要用到：

saved_model_cli show --dir 0/ --all

# 将会输出：
MetaGraphDef with tag-set: 'serve' contains the following SignatureDefs:

signature_def['__saved_model_init_op']:
  The given SavedModel SignatureDef contains the following input(s):
  The given SavedModel SignatureDef contains the following output(s):
    outputs['__saved_model_init_op'] tensor_info:
        dtype: DT_INVALID
        shape: unknown_rank
        name: NoOp
  Method name is:

signature_def['serving_default']:
  The given SavedModel SignatureDef contains the following input(s):
    inputs['input_1'] tensor_info:
        dtype: DT_FLOAT
        shape: (-1, 224, 224, 3)
        name: serving_default_input_1:0
  The given SavedModel SignatureDef contains the following output(s):
    outputs['fc2'] tensor_info:
        dtype: DT_FLOAT
        shape: (-1, 4)
        name: StatefulPartitionedCall:0
  Method name is: tensorflow/serving/predict

可以看到，signature name为“serving_default”，input name为“input_1”，output name为“fc2”。记下来，一会用到。

Docker部署模型

一律使用docker来部署你的模型，如果还不知道docker是什么，不知道怎么用docker来拉取tfs的镜像，请查阅我之前的文章：

Justin ho：TensorFlow Serving + Docker + Tornado机器学习模型生产级快速部署zhuanlan.zhihu.com

这里默认你已经会拉取tfs模型到本地，现在执行以下容器启动命令：

sudo nvidia-docker run -p 8500:8500 
  -v /home/projects/resnet/weights/:/models 
  --name resnet50 
  -itd --entrypoint=tensorflow_model_server tensorflow/serving:2.0.0-gpu 
  --port=8500 --per_process_gpu_memory_fraction=0.5 
  --enable_batching=true --model_name=resnet --model_base_path=/models/season &

这里涉及的参数意义一律看上面那篇文章，这里不解释了。

请求客户端

模型部署起来后，我们要写一个grpc客户端来请求模型，代码参考：

from tensorflow_serving.apis import predict_pb2
from tensorflow_serving.apis import prediction_service_pb2_grpc
import grpc

def request_server(img_resized, server_url):
    '''
    用于向TensorFlow Serving服务请求推理结果的函数。
    :param img_resized: 经过预处理的待推理图片数组，numpy array，shape：(h, w, 3)
    :param server_url: TensorFlow Serving的地址加端口，str，如：'0.0.0.0:8500' 
    :return: 模型返回的结果数组，numpy array
    '''
    # Request.
    channel = grpc.insecure_channel(server_url)
    stub = prediction_service_pb2_grpc.PredictionServiceStub(channel)
    request = predict_pb2.PredictRequest()
    request.model_spec.name = "resnet"  # 模型名称，启动容器命令的model_name参数
    request.model_spec.signature_name = "serving_default"  # 签名名称，刚才叫你记下来的
    # "input_1"是你导出模型时设置的输入名称，刚才叫你记下来的
    request.inputs["input_1"].CopyFrom(
        tf.make_tensor_proto(img_resized, shape=[1, ] + list(img_resized.shape)))
    response = stub.Predict(request, 5.0)  # 5 secs timeout
    return np.asarray(response.outputs["fc2"].float_val) # fc2为输出名称，刚才叫你记下来的

tensorflow 2.0.0请使用以上这段代码，之前那篇tfs的部署文章里面的api已经变了。

我们测试一下，读取一张图片，发送请求到tfs：

from PIL import Image
import numpy as np

imgpath = '20171101110450_48901.jpg'
x = Image.open(imgpath)
x = np.array(x).astype('float32')
x = (x - 128.) / 128.

# grpc地址及端口，为你容器所在机器的ip + 容器启动命令里面设置的port
server_url = '0.0.0.0:8500'  
request_server(x, server_url)

我们将会得到（我这里的resnet只输出4类多标签结果）：

array([0.58116215, 0.04240507, 0.74790353, 0.1388033 ])

完。

宣告：我的B站主页已经悄悄地营业了，来看看我的往期视频：

【铁人贾斯廷】真·AI工程师评测三款steam AI/编程向游戏《While true learn》、《battle bot》、《hacknet》_哔哩哔哩 (゜-゜)つロ干杯~-bilibiliwww.bilibili.com

【硬核自制机器人】终结者or钢铁侠？#如何做一只蠢AI_哔哩哔哩 (゜-゜)つロ干杯~-bilibiliwww.bilibili.com

【飞机起飞】8张泰坦显卡火力全开，来感受下八神泰坦的怒吼！#铁人贾斯廷_哔哩哔哩 (゜-゜)つロ干杯~-bilibiliwww.bilibili.com

【AI】蠢AI的骨架，nvidia jetbot搭建全记录！_哔哩哔哩 (゜-゜)つロ干杯~-bilibiliwww.bilibili.com

本文已收录于本人的个人网站，欢迎浏览收藏：铁人贾斯廷的基地 | 技术宅的碎碎念