使用客户端调用时的疑问:预处理后处理
Created by: iceriver97
我想确认两个事情:
- 对模型输入的预处理与模型输出的后处理,需要自己从原来的脚本中抽离吗? 下面的客户端代码是通用的吗?如果不是我需要修改什么地方?
from paddle_serving_client import Client
from paddle_serving_app.reader import Sequential, File2Image, Resize, Transpose, BGR2RGB, SegPostprocess
import sys
import cv2
import numpy as np
client = Client()
client.load_client_config("seg_client_conf/serving_client_conf.prototxt")
client.connect(["127.0.0.1:9292"])
preprocess = Sequential(
[File2Image(), Resize(
(512, 512), interpolation=cv2.INTER_LINEAR)])
postprocess = SegPostprocess(19)
filename = sys.argv[1]
im = preprocess(filename)
fetch_map = client.predict(feed={"image": im}, fetch=["arg_max_0.tmp_0"])
fetch_map["filename"] = filename
postprocess(fetch_map)
- inference_model_to_serving 函数返回的值是不是我在使用 client.predict()时的 feed 与 fetch的值?