如何将 OpenVino 预测输出转换为框？

Question

我已经使用 openvino/model_server:latest image 部署了 YOLO 模型，并且运行 `import numpy as np 从类导入 imagenet_classes 从 ovmsclient 导入 make_grpc_client

客户端= make_grpc_client（“本地主机：9000”）

打开（“zebra.jpeg”，“rb”）作为f： img = f.read()

output = client.predict({"images": img}, "yolo") print(output.shape)` 这段代码，然后我得到 (1, 84, 8400) 这个输出。如何使用此输出来绘制方框并验证预测的是哪个对象？

Answer 1

根据输出张量结果，您可以使用此 YOLOv8-OpenCV-ONNX-Python 脚本获取边界框，该脚本似乎加载 ONNX 模型（具有相同的输出格式），并将结果转换为具有关联的框分数。