python - Yolo 基于区域的输出到 openvino 的解释脚本

Question

您好，我是 OPENCV/CVAT 的新手，我使用 openvino 运行自动注释，我想使用 YoloV3 执行此任务。

我需要将 Yolo 模型转换为 OpenVINO 格式以用于 opencv/cvat/auto_annotation。 https://github.com/opencv/cvat/tree/develop/cvat/apps/auto_annotation。

要使用自定义模型注释任务，我需要准备 4 个文件：

模型配置 (*.xml) - 带有网络配置的文本文件。
模型权重 (*.bin) - 具有训练权重的二进制文件。
标签映射 (*.json) - 一个带有 label_map 字典的简单 json 文件，类似于带有标签编号字符串值的对象。
解释脚本 (*.py) - 用于将网络输出层转换为可通过 CVAT 处理的预定义结构的文件。此代码将在受限制的 python 环境中运行，但可以使用一些内置函数，如 str、int、float、max、min、range。

我将 Yolo 模型转换为 OpenVINO 格式并创建了 xml 和 bin 文件。我编写映射 lson 文件。现在我需要为 Yolo 的基于区域的输出编写解释 python 脚本。我怎样才能做到这一点？是否有从 tensorflow 模型到 openvino 的中断文件？

score 1 · Accepted Answer

建议您从 Yolo V3 C++ 或 Python 示例开始。

对于 C++：

对于 python 示例：

score 0 · Accepted Answer

这现在与 CVAT 捆绑在一起。解释脚本和 json 文件可以在这里的仓库中找到：https ://github.com/opencv/cvat/tree/develop/utils/open_model_zoo/yolov3

2 回答 2