如何从 GitHub 存储库“杂项”分支安装OpenVINO™训练扩展？

无法从 GitHub* 存储库杂项分支安装 OpenVINO™ 培训扩展。

先决条件

• Ubuntu* 18.04 / 20.04
• Python* 3.6+
• OpenVINO™ - 用于导出和运行模型
• CUDA 工具套件 10.2 - 用于在 GPU 上训练

安装步骤：

1. 下载OpenVINO工具套件 2021.4.2 Linux* 操作系统。
2. 按照 安装和配置 Linux* 版 安装和配置 英特尔® 发行版 OpenVINO™ 工具套件 操作。
3. 克隆 OpenVINO™ 训练扩展存储库：

   git clone https://github.com/openvinotoolkit/training_extensions.git

   export OTE_DIR=`pwd`/training_extensions

4. 克隆 Open Model Zoo 存储库以运行演示：

   git clone https://github.com/openvinotoolkit/open_model_zoo --branch 2021.4.2

   export OMZ_DIR=`pwd`/open_model_zoo

5. 将目录更改为 object_detection：

   cd /training_extensions/models/object_detection

6. 创建 虚拟环境：

   ./init_venv.sh

7. 激活 虚拟环境：

   source venv/bin/activate

8. 在以下 runtime.txt 文件中修改 onnxoptimizer 和 onnx 版本：

   /training_extensions/external/mmdetection/requirements/runtime.txt

   onnx==1.10.1

   onnxoptimizer==0.2.6

9. 安装 在虚拟环境内进行 MMDetection：

   cd /training_extensions/external/mmdetection/

   pip install -r requirements/build.txt

   pip install "git+ https://github.com/open-mmlab/cocoapi.git#subdirectory=pycocotools"

   pip install -v -e .

10. 继续执行 水平文本检测指南中的步骤 2 。

    cd /training_extensions/models/object_detection

    export MODEL_TEMPLATE=`realpath ./model_templates/horizontal-text-detection/horizontal-text-detection-0001/template.yaml`

    export WORK_DIR=/tmp/my-$(basename $(dirname $MODEL_TEMPLATE))

    export SNAPSHOT=snapshot.pth

    python ../../tools/instantiate_template.py ${MODEL_TEMPLATE} ${WORK_DIR}

11. 继续步骤 3。试用预训练的模型

    cd ${WORK_DIR}

    python export.py \

    --load-weights ${SNAPSHOT} \

    --save-model-to export

    python ${OMZ_DIR}/demos/object_detection_demo/python/object_detection_demo.py \

    -m export/model.xml \

    -at ssd \

    -i /dev/video0

故障排除步骤：

如果 ModuleNotFoundError： 没有发生名为 'mmcv._ext' 的模块 ，请将 CUDA 降级到版本 10.2。

如果出现 运行时错误：未找到或未正确配置 OpenVINO™模型优化器 ，请完成第 4 步：配置模型优化器中的步骤。

如果错误仍然存在，请为 /mmdetection/tools/export.py 文件（第 154、171、179 和 185 行）中的 mo.py 提供完整的路径。

mo.py 的完整路径：

/opt/intel/openvino_2021.4.752/deployment_tools/model_optimizer/mo.py