onnxruntime 部署时input注意

 

input_feed中的feats是转换成onnx时的名字，其可通过https://netron.app/  查看onnx模型结构得到。

feats2 = feats2.unsqueeze(0).numpy()
embeddings2 = session.run(
        output_names=['embs'],
        input_feed={
            'feats': feats2
        }
    )

　　

posted @ 2022-11-18 14:48  wieneralan  阅读(127)  评论(0)  编辑  收藏  举报